numpy学习笔记(持续更新)
1.np.where(condition, x, y)
condition表示条件,当条件成立时返回x,否则返回y。可以不设置参数x与参数y,此时返回的是符合条件元素的下标,以元组的形式。
前半段代码对xs进行处理,当第二列数据>43时,将其替换为43。后半段代码不设置x、y参数。
import numpy as npxs = np.array([[1180, 37], [1500, 45]])
#三个参数都设置
xs_processed = xs.copy()
xs_processed[:, 1] = np.where(xs_processed[:, 1] > 43, 43, xs_processed[:, 1])
print(xs_processed)#不设置x、y参数
xss = np.array([12,3,4])
ret = np.where(xss>43)
print(ret)# 输出(0,2)元组