python - Numpy 选择沿轴有条件的元素

Question

我有一个 2D numpy 数组 x 为:

[  [ 1,  2,  3],
   [ 4,  5,  6],
   [ 7,  8,  9],
   [10, 11, 12],
   [13, 14, 15],
   [16, 17, 18],
   [19, 20, 21],
   [22, 23, 24],
   [25, 26, 27],
   [28, 29, 30],
   [31, 32, 33],
   [34, 35, 36],
   [37, 38, 39],
   [40, 41, 42],
   [43, 44, 45],
   [46, 47, 48],
   [49, 50, 51],
   [52, 53, 54],
   [55, 56, 57],
   [58, 59, 60]  ]

我想提取行中任何元素小于 25 的行的参数。因此，我需要输出的是 [0,1,2,3,4,5,6,7]仅针对行，但使用 np.where(x<35)给我所有可能值的二维参数列表。换句话说，我想要的是 x 的所有行的参数，其中至少一个元素小于 25，但我得到的是 x 的所有元素的参数小于 25。我应该怎么办？是否有一个特定的函数，或者我应该从返回的参数列表中选择唯一值？

Answer 1

一种方法是这样的:

import numpy as np
# x is your array
x1 = (x < 25).sum(axis = 1)
rows = np.where(x1 > 0)[0]

行索引位于 rows 中，格式为 array([0, 1, 2, 3, 4, 5, 6, 7])。

您还可以使用非零作为:

rows = np.nonzero((x < 25).sum(axis = 1))[0]