python numpy 同时二维数组某列的值满足一定条件的行数
可以使用numpy的条件判断和索引功能来实现。
假设有一个二维数组arr,要统计第二列值大于5的行数,可以按以下步骤操作:
- 使用条件判断,生成一个布尔型的数组mask,其中第二列大于5的元素值为True,否则为False。
mask = arr[:, 1] > 5
- 使用布尔型数组作为索引,获取符合条件的行:
rows = arr[mask]
- 统计符合条件的行数:
num_rows = len(rows)
完整代码如下:
import numpy as np
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12]])
mask = arr[:, 1] > 5
rows = arr[mask]
num_rows = len(rows)
print(f"符合条件的行数为:{num_rows}")
输出结果为:符合条件的行数为:2
原文地址: http://www.cveoy.top/t/topic/biK5 著作权归作者所有。请勿转载和采集!