因为很多数据是numpy的ndarray格式的,在对ndarray进行处理时就涉及到条件的筛选,比如去掉数据中的小于1或者大于1的数据,而且会经常用到,所以就写个博客记录一下。
1.np.where()
如果用np.where 函数,返回的是符合的元素的index,具体看代码
import numpy as np
x=np.arange(1,10,1)
x=x[np.where(x>4)]
print(x)
2.姑且称为切片法吧
import numpy as np
x=np.arange(1,10,1)
x=x[x>4]
print(x)
显示还是方法二比较简单快捷,代码的可读性也高