Python矩阵查找索引

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

def find_index(matrix, target):
    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            if matrix[i][j] == target:
                return (i, j)
    return None

target = 5
index = find_index(matrix, target)
print(f"元素 {target} 的索引是: {index}")

import numpy as np

matrix = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

def find_index(matrix, target):
    result = np.where(matrix == target)
    if len(result[0]) > 0:
        return tuple(result[0][0]), tuple(result[1][0])
    return None

target = 5
index = find_index(matrix, target)
print(f"元素 {target} 的索引是: {index}")

def find_index(matrix, target):
    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            if matrix[i][j] == target:
                return (i, j)
    raise ValueError(f"元素 {target} 不存在于矩阵中")