python numpy 数组中元素大于等于0的元素

查看原文

总结了四种方式，都没有直接用到循环，希望有用。

>>> import numpy as np>>> a = np.random.randint(-5, 5, (5, 5))>>> aarray([[-4, -4, -5,  2,  1],       [-1, -2, -1,  3,  3],       [-1, -2,  3, -5,  3],       [ 0, -3, -5,  1, -4],       [ 0,  3,  1,  3, -4]])# 方式一>>> np.maximum(a, 0)array([[0, 0, 0, 2, 1],       [0, 0, 0, 3, 3],       [0, 0, 3, 0, 3],       [0, 0, 0, 1, 0],       [0, 3, 1, 3, 0]])# 方式二>>> (a + abs(a)) / 2array([[0, 0, 0, 2, 1],       [0, 0, 0, 3, 3],       [0, 0, 3, 0, 3],       [0, 0, 0, 1, 0],       [0, 3, 1, 3, 0]])# 方式三>>> b = a.copy()>>> b[b < 0] = 0>>> barray([[0, 0, 0, 2, 1],       [0, 0, 0, 3, 3],       [0, 0, 3, 0, 3],       [0, 0, 0, 1, 0],       [0, 3, 1, 3, 0]])# 方式四>>> np.where(a > 0, a, 0)array([[0, 0, 0, 2, 1],       [0, 0, 0, 3, 3],       [0, 0, 3, 0, 3],       [0, 0, 0, 1, 0],       [0, 3, 1, 3, 0]])

试了下：在三维数组时，直接使用b[b < 0] = 0，即可让所有小于0的元素，等于0.

统计所有元素之和：np.sum(arr)