Numpy 数组列表未成功附加所有整数

Question

代码：

import matplotlib.pyplot as plt
import numpy as np

poslist = []
numlist = []
i = 1

num = int(input('Number to check?'))
plt.title(str(num) + ': Collatz Graph')
plt.xlabel('Step')
plt.ylabel('Value')
print(num)
poslist.append(poslist, i)
numlist.append(numlist, num)

while True:
    i = i + 1
    if num % 2 == 0:
        num = num // 2
        print(num)
    elif num == 1:
        break
    else:
        num = num * 3
        num = num + 1
        print(num)
        poslist.append(poslist, i)
        numlist.append(numlist, num)

print('Number Array:', numlist)
print('Position Array', poslist)
plt.plot(*(poslist, numlist))
plt.show()

当我使用listname.append(listname, var) NumPy 函数时，它会跳过大部分变量。它只会将每 3 个或第 4 个变量添加到列表中。对于计步器阵列也是如此。即使添加 time.sleep(4) 函数，它似乎也一次计算 3 或 4 个数字 - 只记录第一个。

Answer 1

似乎有两个问题。

1. 您的append 命令适用于np.array，但poslist 和numlist 是list 类型。对于那些它只是numlist.append(num) 和poslist.append(i)。你想要的是list version，而不是numpy.append。
2. 正如 diggusbiccus 指出的那样，您在 while 循环中的 append 语句位于 if 语句中，您需要将它们推回以使其位于其外部。

import matplotlib.pyplot as plt
#import numpy as np <--- You are not using numpy

poslist = []
numlist = []
i = 1

num = int(input('Number to check?'))
plt.title(str(num) + ': Collatz Graph')
plt.xlabel('Step')
plt.ylabel('Value')
print(num)
poslist.append(i) # <--- Change append statement for list
numlist.append(num) # <--- Same as above

while True:
    i = i + 1
    if num % 2 == 0:
        num = num // 2
        print(num)
    elif num == 1:
        break
    else:
        num = num * 3
        num = num + 1
        print(num)
    poslist.append(i) # <--- Shift out of if/else statement and change append
    numlist.append(num) # <--- Same as above

print('Number Array:', numlist)
print('Position Array', poslist)
plt.plot(*(poslist, numlist))
plt.show()

如果您想使用 numpy 数组，另一个选择是更改您的列表。如果想使用np.append，您可以使用poslist = np.append(poslist, i) 之类的东西。但是，正如 hpaulj 对此答案的 cmets 所建议的那样，如果您想使用np.concatenate，它看起来更像poslist = np.concatenate((poslist, [i,]))。 （This post 比较这两种方法。）

import matplotlib.pyplot as plt
import numpy as np 

poslist = np.array([]) # <--- change list to array
numlist = np.array([]) # <--- change list to array
i = 1

num = int(input('Number to check?'))
plt.title(str(num) + ': Collatz Graph')
plt.xlabel('Step')
plt.ylabel('Value')
print(num)

#poslist = np.append(poslist, i) # <--- Fix syntax of np.append
poslist = np.concatenate((poslist, [i,])) # <-- Or use concatenate instead
numlist = np.concatenate((numlist, [num,])) # <-- Or use concatenate instead

while True:
    i = i + 1
    if num % 2 == 0:
        num = num // 2
        print(num)
    elif num == 1:
        break
    else:
        num = num * 3
        num = num + 1
        print(num)
    poslist = np.concatenate((poslist, [i,])) # <--- Move out of if/else statement
    numlist = np.concatenate((numlist, [num,])) 

print('Number Array:', numlist)
print('Position Array', poslist)
plt.plot(*(poslist, numlist))
plt.show()