Python错误，“NumPy布尔数组索引赋值需要0或1维输入，输入有2维”

Question

我对python有点陌生，目前正在从事一个项目，并且在这几行代码中遇到了这个错误

    g1_coll[obstacle==0]=tau*(g1+g2-g3+g4)
    g2_coll[obstacle==0]=tau*(g1+g2+g3-g4)
    g3_coll[obstacle==0]=tau*(-g1+g2+g3+g4)
    g4_coll[obstacle==0]=tau*(g1-g2+g3+g4)

有人能帮我理解吗

Answer 1

我假设你得到的错误是因为你所有的数组都是二维的。我建议您尝试使用numpy.putmask(matrix, mask, new_matrix_values)

比如说

    mask = (obstacle == 0)
    numpy.putmask(g1_coll, mask, tau*(g1+g2-g3+g4))
    numpy.putmask(g2_coll, mask, tau*(g1+g2+g3-g4))
    numpy.putmask(g3_coll, mask, tau*(-g1+g2+g3+g4))
    numpy.putmask(g4_coll, mask, tau*(g1-g2+g3+g4))