温馨提示×

python col查找特定值的方法

小樊
81
2024-09-12 17:20:56
栏目: 编程语言

在Python中,可以使用列表推导式(list comprehension)或循环来查找特定值。

假设我们有一个二维列表(矩阵)matrix,我们想要找到值为target的所有元素的索引。

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

target = 5

方法1:使用列表推导式

result = [(i, j) for i, row in enumerate(matrix) for j, value in enumerate(row) if value == target]
print(result)  # 输出:[(1, 1)]

方法2:使用循环

result = []
for i, row in enumerate(matrix):
    for j, value in enumerate(row):
        if value == target:
            result.append((i, j))
print(result)  # 输出:[(1, 1)]

这两种方法都会返回一个包含值为target的元素索引的列表。在这个例子中,值为5的元素的索引是(1, 1)。

0