Question

如何获得满足指定条件的numpy 2d数组的行号和列号？例如，我有一个2d数组（所有浮点数），我想获得最小值或最大值所在的位置（行和列索引）。

Answer 1

您可以使用np.where()，如以下示例所示：

In [46]: arr = np.arange(10, dtype=float32).reshape(5, 2)

In [47]: arr
Out[47]: 
array([[ 0.,  1.],
       [ 2.,  3.],
       [ 4.,  5.],
       [ 6.,  7.],
       [ 8.,  9.]], dtype=float32)

# get row and column index of minimum value in arr
In [48]: np.where(arr == arr.min())
Out[48]: (array([0]), array([0]))

# get the indices of maximum element in arr
In [49]: np.where(arr == arr.max())
Out[49]: (array([4]), array([1]))

即使数组中有多个最小值/最大值，上述方法也可以正常工作。

In [59]: arr
Out[59]: 
array([[ 0.,  0.],
       [ 2.,  3.],
       [ 4.,  5.],
       [ 6.,  7.],
       [ 9.,  9.]], dtype=float32)

In [60]: np.where(arr == arr.max())
Out[60]: (array([4, 4]), array([0, 1]))  # positions: (4,0) and (4,1)

In [61]: np.where(arr == arr.min())
Out[61]: (array([0, 0]), array([0, 1]))  # positions: (0,0) and (0,1)

如何获得满足指定条件的numpy 2d数组的行号和列号？

1 个答案: