可以使用numpy的条件判断和索引功能来实现。

假设有一个二维数组arr,要统计第二列值大于5的行数,可以按以下步骤操作:

  1. 使用条件判断,生成一个布尔型的数组mask,其中第二列大于5的元素值为True,否则为False。
mask = arr[:, 1] > 5
  1. 使用布尔型数组作为索引,获取符合条件的行:
rows = arr[mask]
  1. 统计符合条件的行数:
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

python numpy 同时二维数组某列的值满足一定条件的行数

原文地址: http://www.cveoy.top/t/topic/biK5 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录