numpy实现sumif

可以使用numpy的where函数实现sumif，示例如下：

import numpy as np

# 生成一个随机数组
arr = np.random.randint(0, 10, size=(5, 5))
print(arr)

# 统计arr中大于等于5的元素之和
sum = np.sum(np.where(arr >= 5, arr, 0))
print(sum)

输出结果如下：

[[2 0 9 2 8]
[8 1 7 0 2]
[2 8 3 3 3]
[9 9 3 7 4]
[4 4 4 4 4]]
43