python - 如何遍历 numpy 3d 数组

Question

所以我有一个数组：

array([[[27, 27, 28],
        [27, 14, 28]],

       [[14,  5,  4],
        [ 5,  6, 14]]])

我如何遍历它并在每次迭代时获取 [a, b, c] 值，我尝试这样：

for v in np.nditer(a):
    print(v)

但它只是打印

27
27
28
27
14
28
14
5
4
5
6

我需要：

[27 27 28]
[27 14 28]...

标签： pythonloopsnumpy

Answer 1

b = a.reshape(-1, 3)
for triplet in b:
    ...