我希望您可以把一个数组中所有不在另一个数组中的值都设为0。
如果只有一个条件,那么很容易实现:
如果我有一系列条件,例如:
但我无法找到一般的做法,类似于这样。
如果只有一个条件,那么很容易实现:
a = np.array([[1,2],[2,4],[5,6]])
cond = 1
a[a!=cond] = 0
如果我有一系列条件,例如:
cond = np.array([1,2,6])
我可以像这样写出来
。a[(a!=1) & (a!=2) & (a!=6)]=0
但我无法找到一般的做法,类似于这样。
a[a!=cond] = 0
当cond
是一个数组时。我也看了np.select
,但那似乎不是我需要的。