查找被分成9个3x3数组的9x9数组的元素的索引

def find_element_index(matrix, target):
    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            if matrix[i][j] == target:
                # 计算小数组索引
                small_array_index = (i // 3) * 3 + (j // 3)
                # 计算小数组内部元素索引
                small_array_element_index = (i % 3) * 3 + (j % 3)
                return i, j, small_array_index, small_array_element_index
    return None

# 示例输入
matrix = [
    [5, 3, 0, 0, 7, 0, 0, 0, 0],
    [6, 0, 0, 1, 9, 5, 0, 0, 0],
    [0, 9, 8, 0, 0, 0, 0, 6, 0],
    [8, 0, 0, 0, 6, 0, 0, 0, 3],
    [4, 0, 0, 8, 0, 3, 0, 0, 1],
    [7, 0, 0, 0, 2, 0, 0, 0, 6],
    [0, 6, 0, 0, 0, 0, 2, 8, 0],
    [0, 0, 0, 4, 1, 9, 0, 0, 5],
    [0, 0, 0, 0, 8, 0, 0, 7, 9]
]
target = 3

# 调用函数进行查找
result = find_element_index(matrix, target)

# 输出结果
if result:
    row, col, small_array_index, small_array_element_index = result
    print("目标元素的索引：")
    print("行数：", row)
    print("列数：", col)
    print("小数组索引：", small_array_index)
    print("小数组内部元素索引：", small_array_element_index)
else:
    print("未找到目标元素")