以下是用Python编写的按行查找满足2D数组条件的元素的索引的代码示例:
def find_elements(matrix, condition):
result = []
for i in range(len(matrix)):
for j in range(len(matrix[i])):
if condition(matrix[i][j]):
result.append((i, j))
return result
# 示例使用
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
# 查找偶数元素的索引
even_indices = find_elements(matrix, lambda x: x % 2 == 0)
print(even_indices) # 输出: [(0, 1), (1, 2), (2, 1)]
# 查找大于5的元素的索引
greater_than_5_indices = find_elements(matrix, lambda x: x > 5)
print(greater_than_5_indices) # 输出: [(2, 0), (2, 1), (2, 2)]
这个示例中的find_elements
函数接受一个二维数组matrix
和一个条件函数condition
作为参数。它遍历整个二维数组,对每个元素应用条件函数,并将满足条件的元素的索引添加到结果列表中。最后返回结果列表。
在示例中,我们使用了两个条件函数:一个用于查找偶数元素的索引,另一个用于查找大于5的元素的索引。你可以根据具体的条件函数来修改代码以满足你的需求。
上一篇:按行查询索引