如何迭代 numpy 数组，每个循环获取两个值？

Question

我设想类似的东西

import numpy as np
x = np.arange(10)
for i, j in x:
     print(i,j)

得到类似的东西

0 1
2 3
4 5
6 7
8 9

但我得到了这个追溯：

Traceback (most recent call last):
  File "/home/andreas/.local/share/JetBrains/Toolbox/apps/PyCharm-P/ch-0/223.8214.51/plugins/python/helpers/pydev/pydevconsole.py", line 364, in runcode
    coro = func()
  File "<input>", line 1, in <module>
TypeError: cannot unpack non-iterable numpy.int64 object

我还尝试将np.nditer(x)和itertools与zip(x[::2], x[1::2])一起使用，但这也不起作用，并出现不同的错误消息。

这应该超级简单，但我无法在网上找到解决方案。

Answer 1

您试图将0放入i和j ，这是不可能的。 要实现该结果，您必须使用x = x.reshape((5,2))或x.shape = 5, 2重塑 numpy 数组。 然后你可以像这样解压它。

为了形象化，这就是您当前的代码正在做的事情：

i, j = 0
...
i, j = 1
...

如果你重塑它，这就是会发生的事情：

i, j = [0, 1]
...
i, j = [2, 3]
...

编辑：

import numpy as np
N = 10

x = np.arange(N).reshape((N/2, 2))
for i, j in x:
     print(i,j)

Answer 2

这是你想要的吗？

for x in range(0,10,2):
    print(x, x+1)

或者您的值保存在 numpy 数组 x 中，然后

for i in range(0, len(x), 2):
    print(x[i],x[i+1])

Answer 3

试图忠实于最初的尝试。 zip ping 偶数和奇数对的元组：

import numpy as np
x = np.arange(10)

for i, j in zip(x[::2], x[1::2]):
     print(i,j)