筛选计数,
统计第2列中大于0的个数。
方法1:
if __name__ == '__main__':
import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("count2",np.sum(data[:, 1] > 2))
print("sum2",np.sum(data[:, 1][data[:, 1] > 2]))
方法2:
if __name__ == '__main__':
import numpy as np
data = np.array([[1,2,3],[4,5,6],[7,8,9]])
result=np.where(data[:,1]>0)
print(data[:,1])
print(result[0].shape[0])
筛选之后求和:
if __name__ == '__main__':
import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
result = np.where(data[:, 1] > 2)
print(data[:, 1])
print(result[0].shape[0])
select_data=data[:, 1][result]
print(np.sum(select_da