如何使用用户输入修改列表(2D数组)？(Python)

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

def modify_2d_array(matrix):
    try:
        # 获取用户输入的行索引、列索引和新值
        row_index = int(input("请输入要修改的元素的行索引: "))
        col_index = int(input("请输入要修改的元素的列索引: "))
        new_value = int(input("请输入新的值: "))
        
        # 验证索引是否有效
        if 0 <= row_index < len(matrix) and 0 <= col_index < len(matrix[0]):
            # 修改数组元素
            matrix[row_index][col_index] = new_value
            print("修改成功！")
        else:
            print("索引超出范围，请输入有效的索引。")
    except ValueError:
        print("请输入整数类型的值。")
    except IndexError:
        print("数组索引出错。")

# 示例二维数组
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 显示原始数组
print("原始数组:")
for row in matrix:
    print(row)

# 修改数组
modify_2d_array(matrix)

# 显示修改后的数组
print("修改后的数组:")
for row in matrix:
    print(row)