Python:在多维列表中按范围查找索引

def find_indexes_in_range(matrix, min_value, max_value):
    indexes = []
    for i, row in enumerate(matrix):
        for j, value in enumerate(row):
            if min_value <= value <= max_value:
                indexes.append((i, j))
    return indexes

# 示例二维列表
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 查找值在2到8之间的元素的索引
indexes = find_indexes_in_range(matrix, 2, 8)
print(indexes)  # 输出: [(0, 1), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1)]

import numpy as np

def find_indexes_in_range_numpy(array, min_value, max_value):
    mask = (array >= min_value) & (array <= max_value)
    return np.argwhere(mask)

# 示例二维NumPy数组
array = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

# 查找值在2到8之间的元素的索引
indexes = find_indexes_in_range_numpy(array, 2, 8)
print(indexes)  # 输出: [[0 1] [1 0] [1 1] [1 2] [2 0] [2 1]]