Numpy.ndarray：按數組的特定軸進行迭代

Question

我有一個多維數組。 而且我必須通過其某些軸進行迭代。 如果需要所有軸，則可以使用nditer ，但是如果僅需要特定軸，則必須手動執行：

my_array = np.arange(3 * 4 * 5).reshape((3, 4, 5))
for i in range(my_array.shape[0]):
    for j in range(my_array.shape[1]):
        print(i, j)
        # Here should be some processing of the 3rd dimension items of the (i,j)

您不能建議我一個更簡單的方法嗎？

Answer 1

考慮傳遞到單個循環，並使用ndindex （ docs ）：

my_array = np.arange(3 * 4 * 5).reshape((3, 4, 5))
for ij in np.ndindex(my_array.shape[:2]):
    i,j=ij
    print(i,j)