统计一个numpy数组里面数字1的个数
时间: 2024-03-14 13:44:17 浏览: 124
可以使用numpy库中的count_nonzero函数来统计一个numpy数组中数字1的个数,示例如下:
```python
import numpy as np
arr = np.array([1, 2, 1, 3, 4, 1])
count = np.count_nonzero(arr == 1)
print(count) # 输出3
```
在这个示例中,我们首先创建了一个numpy数组arr,然后使用`arr == 1`得到一个布尔型的数组,表示arr中每个元素是否等于1。最后,我们使用`np.count_nonzero`函数统计了这个布尔型数组中True的个数,也就是数组中数字1的个数。
相关问题
编写python函数,要求:输入a,b,c,d,n分别是均值,方差,最大值,最小值,个数。要求输出均值为a,方差为b,最小值为c,最大值为d,共e个数字的符合正态分布数组
好的,您的问题已经收到了。我来回答您的问题。
您需要编写以下Python函数:
```
import numpy as np
def generate_norm_distribution(a, b, c, d, n):
# 生成均值为a,方差为b的正态分布数组
arr = np.random.normal(a, np.sqrt(b), n)
# 限制数组元素的范围为c到d
arr = np.clip(arr, c, d)
# 统计数组元素的个数
e = arr.size
# 打印结果
print("均值为:{}".format(a))
print("方差为:{}".format(b))
print("最小值为:{}".format(c))
print("最大值为:{}".format(d))
print("共{}个数字的符合正态分布数组为:{}".format(e, arr))
return arr
```
这个函数输入了五个参数:均值a,方差b,最大值d,最小值c,以及生成的数组元素个数n。这个函数的输出包括了均值,方差,最小值,最大值,以及符合正态分布的数组,数组大小为n。函数通过numpy模块进行实现。
您可以通过以下方式调用本函数:
```
generate_norm_distribution(0, 1, -1, 1, 10)
```
注意:参数a和d需要满足条件a<d。否则生成的数组无法满足最小值和最大值的限制条件。
阅读全文