今天看到了一种新用法,记录一下
代码如下:
a = []
text = np.random.randint(1,5,[4,5])
for i in range(text.shape[0]):
a.append(np.sum(1 == text[i]))
这里的会输出什么呢?
text:
[[4 2 3 2 1]
[2 4 3 2 2]
[4 3 2 4 2]
[2 2 1 1 2]]
a:
[1, 0, 0, 2]
输入的是有行数个数组的一维数组 记录的是每一行是1的数量
原来np.sum()还可以这样用,有时候确实挺方便