如何返回二维数组中除一列以外的所有列？

def remove_column(arr, column_index):
    new_arr = []
    for row in arr:
        new_row = []
        for i, val in enumerate(row):
            if i != column_index:
                new_row.append(val)
        new_arr.append(new_row)
    return new_arr

# 示例用法
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
column_index = 1
result = remove_column(arr, column_index)
print(result)

[[1, 3], [4, 6], [7, 9]]