我正在python上构建一个矩阵，一旦它被构建，我想删除所有只包含零的列。（有些列包含零，但不仅仅是因为我想保留它们。）

def remove_column_with_all_zeros(matrix):
    zero_columns = []
    for i in range(len(matrix[0])):
        column = [row[i] for row in matrix]
        if all(val == 0 for val in column):
            zero_columns.append(i)
    for i in sorted(zero_columns, reverse=True):
        for row in matrix:
            del row[i]
    return matrix

我试过这个功能了，但是不起作用。
谢谢