如何修改Numpy代码实现数组行阈值筛选的两类定制输出？

我有一个形状为(10,5)的numpy数组Y，当前能检查每行中大于阈值的元素及其位置，但需要实现两个定制需求：

1. 若某行中不存在大于阈值的元素，需输出该行索引与None；
2. 若某行中有多个大于阈值的元素，需输出其中的最大值及其对应位置。

数组Y的具体内容：

import numpy as np
Y = np.array([
    [0.01134, 0.09777, 0.6773, 0.20182, 0.01178],
    [0.08211, 0.35025, 0.10659, 0.36319, 0.09785],
    [0.06689, 0.50127, 0.16266, 0.17762, 0.09156],
    [0.11849, 0.43602, 0.3991, 0.01871, 0.02768],
    [0.03238, 0.68228, 0.27775, 0.00638, 0.0012],
    [0.79637, 0.04201, 0.11199, 0.0028, 0.04684],
    [0.57715, 0.14894, 0.26596, 0.00425, 0.0037],
    [0.38991, 0.31468, 0.2895, 0.00269, 0.00322],
    [0.16056, 0.56997, 0.25396, 0.01414, 0.00137],
    [0.00005, 0.93875, 0.04922, 0.01175, 0.00024]
])

当前使用的代码：

print("Values greater than the threshold:", Y[Y > 0.40])
print("Their positions:", np.argwhere(Y > 0.40))
# Threshold here is 0.40

当前输出结果：

[[ 0  2]
[ 2  1]
[ 3  1]
[ 4  1]
[ 5  0]
[ 6  0]
[ 8  1]
[ 9  1]]

可以看到，行1和行7中没有大于0.40的元素，输出中无对应条目，这类场景需要输出[1, None]和[7, None]；另外，若某行存在多个大于阈值的元素（如示例行[0.16056, 0.56997, 0.45396, 0.01414, 0.00137]），需要输出其中最大值（0.56997）及其位置。请问需对代码做哪些修改才能满足这些条件？

可以通过遍历数组的每一行，针对每行单独处理来满足需求，具体代码如下：

import numpy as np

Y = np.array([
    [0.01134, 0.09777, 0.6773, 0.20182, 0.01178],
    [0.08211, 0.35025, 0.10659, 0.36319, 0.09785],
    [0.06689, 0.50127, 0.16266, 0.17762, 0.09156],
    [0.11849, 0.43602, 0.3991, 0.01871, 0.02768],
    [0.03238, 0.68228, 0.27775, 0.00638, 0.0012],
    [0.79637, 0.04201, 0.11199, 0.0028, 0.04684],
    [0.57715, 0.14894, 0.26596, 0.00425, 0.0037],
    [0.38991, 0.31468, 0.2895, 0.00269, 0.00322],
    [0.16056, 0.56997, 0.25396, 0.01414, 0.00137],
    [0.00005, 0.93875, 0.04922, 0.01175, 0.00024]
])

threshold = 0.40

for row_idx, row in enumerate(Y):
    # 筛选当前行中大于阈值的元素及其索引
    mask = row > threshold
    valid_elements = row[mask]
    valid_indices = np.where(mask)[0]
    
    if len(valid_elements) == 0:
        print(f"[{row_idx}, None]")
    else:
        # 找到最大值对应的列位置
        max_val_idx_in_row = valid_indices[np.argmax(valid_elements)]
        max_val = valid_elements.max()
        print(f"[{row_idx}, ({max_val_idx_in_row}, {max_val})]")

代码逻辑说明 #

• 用enumerate遍历数组，同时获取行索引和行数据
• 布尔掩码筛选出当前行中符合阈值要求的元素和它们在该行的位置
• 无符合条件元素时，直接输出[行索引, None]
• 有符合条件元素时，找到最大值对应的列位置，输出[行索引, (列位置, 最大值)]

运行输出 #

[0, (2, 0.6773)]
[1, None]
[2, (1, 0.50127)]
[3, (1, 0.43602)]
[4, (1, 0.68228)]
[5, (0, 0.79637)]
[6, (0, 0.57715)]
[7, None]
[8, (1, 0.56997)]
[9, (1, 0.93875)]

内容的提问来源于stack exchange，提问作者A. Gehani