MATLAB语句A（B==1）=C的python等价物

Question

我有三个numpy数组，如下所示：

A = [1, 2, 3, 4, 5]
B = [0, 1, 0, 0, 1]
C = [30, 40]

我想替换A的元素，它们在B中的等效值等于1。对于上面的例子，我想得到：

A = [1, 30, 3, 4, 40]

在MATLAB中，我可以做到：

A(B==1) = C'

你知道Python中的一个等价代码吗（最好是在A和B也是多维的情况下工作的代码）？提前谢谢。你知道吗

Answer 1

语法非常相似：

>>> import numpy as np
>>> A = np.array([1, 2, 3, 4, 5])
>>> B = np.array([0, 1, 0, 0, 1])
>>> C = np.array([30, 40])
>>> A[B==1] = C
>>> A
array([ 1, 30,  3,  4, 40])