如何仅对满足索引内条件的二维 numpy 数组的元素执行操作，而不在 Python 中迭代？

Question

我正在尝试实现一种求解算法，如果数组元素为负，则使用一个方程，如果为正，则使用另一个方程。 这是我试图区分这两种情况的代码：

    Xn = X.copy()
    X[X[2:-2, 2:-2]>=0] = #algorithm A, calls several upstream values from Xn
    X[X[2:-2, 2:-2]<0] = #algorithm B, calls several downstream values from Xn

我只想为 X[2:-2, 2:-2] 赋值。 X 的形状是 (21,21) 而 X[2:-2,2:-2] 是 (17,17)。 我用上面的代码得到这个错误：

IndexError：boolean 索引与维度 0 上的索引数组不匹配； 尺寸为 21，但对应的 boolean 尺寸为 17

在不迭代每个元素的情况下如何做到这一点？

Answer 1

使用矢量化。 这将解决你所有的问题。