不定期更新,多积累
where
语法格式
numpy.where(condition[, x, y])
位置函数,通过条件筛选符合要求的数据。...若是满足就用xxx来填充,不满足就用yyy来填充
案例展示
import numpy as np
a = np.array([[1,2,3,4],[3,4,5,6],[6,7,8,9]])
a
array...([[1, 2, 3, 4],
[3, 4, 5, 6],
[6, 7, 8, 9]])
# 返回大于5的位置
np.where(a > 5)
(array([1, 2,...2, 2, 2]), array([3, 0, 1, 2, 3]))
# 大于5的填充为1,小于或等于5的填充为10
np.where(a > 5,1,10)
array([[10, 10, 10