从二维数组Python中提取索引

def find_index(arr, target):
    for i in range(len(arr)):
        for j in range(len(arr[i])):
            if arr[i][j] == target:
                return i, j
    return -1, -1  # 如果未找到目标元素，返回-1

# 示例用法
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
target = 5
row, col = find_index(arr, target)
print(f"目标元素 {target} 的索引为：({row}, {col})")

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
target = 5
indices = np.where(arr == target)
row, col = indices[0][0], indices[1][0]
print(f"目标元素 {target} 的索引为：({row}, {col})")