如何创建2D数组作为array.reshape()方法的重写代码

def reshape_array(arr, shape):
    # 验证元素数量是否匹配
    if len(arr) != shape[0] * shape[1]:
        return "无法重塑数组，元素数量不匹配"

    # 创建新的2D数组
    reshaped_arr = []
    index = 0

    # 按照目标形状重新排列元素
    for i in range(shape[0]):
        row = []
        for j in range(shape[1]):
            row.append(arr[index])
            index += 1
        reshaped_arr.append(row)

    return reshaped_arr

# 原始数组
original_arr = [1, 2, 3, 4, 5, 6]

# 目标形状
target_shape = (2, 3)

# 调用重写的reshape_array方法
result = reshape_array(original_arr, target_shape)

print(result)

[[1, 2, 3], [4, 5, 6]]