numpy的花式操作

1. np.where & np.in1d

满足条件(condition),输出x,不满足输出y。
np.where(condition, x, y)
# ar1中的元素是否出现在arr2,返回一个bool矩阵
np.in1d(ar1, ar2, assume_unique=False, invert=False)
np.where(np.in1d(x,y),x,y)

猜你喜欢

转载自www.cnblogs.com/yutingmoran/p/11927832.html