Numpy选择具有沿轴条件的元素

我有一个二维 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]仅行，但 usingnp.where(x<35)是为我提供所有可能值的 2D 参数列表。换句话说，我想要的是 x 的所有行的参数，其中至少有一个元素小于 25，但我得到的是 x 的所有元素小于 25 的参数。我应该怎么做做？是否有一个特定的函数，或者我应该从返回的参数列表中选择唯一的值？

回答

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

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