在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)。