在python中迭代numpy数组行

Question

我正在尝试使用以下 for 循环遍历 python 中两个 numpy 数组的行：

for i, j in range(X.shape[0]), range(y.shape[0]):

但我收到以下错误：

ValueError: too many values to unpack (expected 2)

我认为通过使用每个数组中的行数创建列表，我可以使用 i 的值遍历 X，并使用 j 的值遍历 y。

有人能解释一下为什么这不起作用，以及我怎样才能让它起作用吗？谢谢！

Answer 1

这不是你想要做的。

要遍历 X 中的行和 Y 中的行，您应该使用嵌套循环：

for i in range(X.shape[0]):
    for j in range(Y.shape[0]):
        func(i, j)

话虽如此，我强烈建议您将循环作为最后的手段。尽可能尝试向量化函数。

如果您必须循环，则可以使用numba 来提高性能。