您正在使用python列表而不是numpy数组。尝试以下方法:
import numpy as npm1 = np.array([0,1,1,3])m2 = np.array([0,0,1,1])mask = (m1 == 1) & (m2 == 1)data[mask]# returns array([30])
在您的示例中,当
m1是列表时,
m1 == 1被评估为
False(与相同
m2),因此mask为
False和
data[False] =data[0] = 10。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)