使用nditer迭代两个numpy 2d矩阵

Question

我正在尝试迭代两个numpy矩阵，其中一个大小为nx3，另一个大小为nx1

我试图让nditer同时迭代他们的行。

因此，如果我们有：

y = np.array([   [ 1],
                 [-1],
                 [ 1]   ])
x = np.array([[ 1.3432504 , -1.3311479 ,  1.        ],
             [ 1.8205529 , -0.6346681 ,  1.         ],
             [ 0.98632067, -1.8885762 ,  1.         ]])

我尝试：

for (a,b) in iterator:
   print(a)
   print(b)

这应该给

[1]
[ 1.3432504 , -1.3311479 ,  1.        ]
[-1]
[ 1.8205529 , -0.6346681 ,  1.         ]
[1]
[ 0.98632067, -1.8885762 ,  1.         ]

我用'external_loop flag'尝试了np.nditer，我得到了x的期望输出，但是当我一次只想要一个元素时，它会迫使y分成3个元素。

先感谢您。

Answer 1

您可以使用zip()内置函数：

In [22]: for i, j in zip(y, x):
             print(i);print(j)
   ....:     
[1]
[ 1.3432504 -1.3311479  1.       ]
[-1]
[ 1.8205529 -0.6346681  1.       ]
[1]
[ 0.98632067 -1.8885762   1.        ]

Answer 2

使用nditer很难控制迭代的“深度”。 例如最简单的情况：

In [35]: for i,j in np.nditer([y, x]):
    ...:     print(i, j)
    ...:     
1 1.3432504
1 -1.3311479
1 1.0
-1 1.8205529
-1 -0.6346681
-1 1.0
1 0.98632067
1 -1.8885762
1 1.0

这(i,j) pair of values for each broadcastable combination of y and x的(i,j) pair of values for each broadcastable combination of创建一对(i,j) pair of values for each broadcastable combination of . x is (3, 3), y is (3,1) ，结果是（3,3）.flat上的迭代。

（如果x为np.arange(n) ，则nditer会生成（3,1）数组与（1，n），即（3，n）集的所有组合。

您不能轻易告诉nditer仅在x的行上进行迭代。 external_loop可以做到，但不可预测。

ndindex生成给定深度的索引，但是它是通过创建正确形状的数组来实现的。

In [38]: for i,j in np.ndindex(2,3):
    ...:     print(i,j)
    ...:     
0 0
0 1
0 2
1 0
1 1
1 2

或迭代2个数组的行：

In [39]: for i in np.ndindex(3):
    ...:     print(y[i], x[i,:])
    ...:     
[1] [[ 1.3432504 -1.3311479  1.       ]]
[-1] [[ 1.8205529 -0.6346681  1.       ]]
[1] [[ 0.98632067 -1.8885762   1.        ]]

但是您也可以for i in range(3):很好地使用。

np.nditer是开发cython或其他c-api代码的垫脚石。 nditer的c-api版本具有nditer的功能，并且速度相对较好。 python等效项既不快速也不强大。

您的评论提到了与apply_along_axis进一步接口。 那是用Python编写的，并使用ndindex为需要迭代的轴生成索引。 它可以使某些任务更方便，但不会加快代码的速度。