Question

我是NumPy的新手，我遇到了在numpy数组上运行一些条件语句的问题。假设我有3个看起来像这样的numpy数组：

一个：

[[0, 4, 4, 2],
 [1, 3, 0, 2],
 [3, 2, 4, 4]]

B：

[[6, 9, 8, 6],
 [7, 7, 9, 6],
 [8, 6, 5, 7]]

和，c：

[[0, 0, 0, 0],
 [0, 0, 0, 0],
 [0, 0, 0, 0]]

我有a和b的条件语句，其中我想使用b的值（如果满足a和b的条件）来计算c的值：

c[(a > 3) & (b > 8)]+=b*2

我收到错误说：

Traceback (most recent call last):
  File "<interactive input>", line 1, in <module>
ValueError: non-broadcastable output operand with shape (1,) doesn't match the broadcast shape (3,4)

知道我怎么能做到这一点吗？

我希望c的输出看起来如下：

[[0, 18, 0, 0],
 [0, 0, 0, 0],
 [0, 0, 0, 0]]

Answer 1

您可以使用numpy.where：

np.where((a > 3) & (b > 8), c + b*2, c)
#array([[ 0, 18,  0,  0],
#       [ 0,  0,  0,  0],
#       [ 0,  0,  0,  0]])

或算术：

c + b*2 * ((a > 3) & (b > 8))
#array([[ 0, 18,  0,  0],
#       [ 0,  0,  0,  0],
#       [ 0,  0,  0,  0]])

Answer 2

问题是掩盖接收部分，但不掩盖发件人部分。结果：

return

尺寸不一样。如果您想执行元素添加（基于您的示例），您只需将切片添加到右侧部分：

c[(a > 3) & (b > 8)]+=b*2
# ^ 1x1 matrix        ^3x4 matrix

或提高效率：

c[(a > 3) & (b > 8)]+=b[(a > 3) & (b > 8)]*2

Answer 3

numpy表达式的微小变化将得到预期的结果：

c += ((a > 3) & (b > 8)) * b*2

这里首先我使用((a > 3) & (b > 8))的布尔值创建一个蒙版矩阵，然后将矩阵乘以b*2，然后生成一个3x4矩阵，可以很容易地添加到c 1}}

numpy数组的条件运算

3 个答案: