求矩阵中数值为某一值的元素个数的方法
时间: 2023-07-16 22:16:44 浏览: 84
matlab 计算矩阵中不同元素个数的函数
4星 · 用户满意度95%
可以使用以下方法求矩阵中数值为某一值的元素个数:
1. 遍历矩阵中的每个元素,计算出与目标值相等的元素数量。
2. 利用numpy库函数count_nonzero(),统计矩阵中与目标值相等的元素数量。
下面是具体的代码实现:
```python
import numpy as np
# 生成一个 3 x 3 的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 统计矩阵中值为 5 的元素个数
count = np.count_nonzero(matrix == 5)
print(count) # 输出 1
```
在上面的代码中,我们首先使用numpy库生成了一个3x3的矩阵。然后使用`matrix == 5`计算出一个布尔数组,其中为True的元素表示矩阵中与目标值相等的元素。最后,使用`np.count_nonzero()`函数统计这个布尔数组中True的元素数量,即为矩阵中与目标值相等的元素数量。
阅读全文