我是一个 NumPy 新手,我在运行一些对 NumPy 数组的条件语句时遇到了问题。假设我有三个类似于以下内容的 NumPy 数组:
a:
[[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的条件语句,如果满足a和b的条件,我想用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]]