使用nditer的浅层迭代

Question

我有这样一个数组：

>>>y = np.random.randint(0, 255, (2,2,3))
>>>array([[[242, 14, 211],
           [198,  7,   0]],

          [[235,  60,  81],
           [164,  64, 236]]])

而且我必须遍历每个 三元组 元素（不幸的是，矢量化在这里对我没有帮助…）。所以我尝试了：

for i, j in np.nditer(y):
print y[i, j],

希望我能得到这样的输出：

[242, 14, 211], [198, 7, 0], [235, 60, 81], [164, 64, 236]，但是没有运气！

我得到错误：

Traceback (most recent call last):

  File "<ipython-input-21-a336ef837a8a>", line 1, in <module>
    for i, j in np.nditer(y):    print y[i,j]

TypeError: iteration over a 0-d array

我敢肯定我犯了一个非常明显的错误……有人可以帮我吗？

Answer 1

看来您只需要将其展平。您可以chain从itertools使用运算符。

from itertools import chain


y = np.random.randint(0, 255, (2,2,3)
b = chain.from_iterable(y) # where b is a generator

清单（b）输出

[array([ 51, 119,  84]),
 array([ 50, 110, 193]),
 array([165, 157,  52]),
 array([239, 119,  83])]