在Python中删除列

Question

有一个用逗号分隔的文件，其中包含约50列和几行，因此我需要删除所有始终为0的列（即该列中的所有值为零）。

使用以下代码读取文件：

with open('data.txt', 'rb') as f:
    reader.csv.reader(f, delimiter=',')
    for row in reader:
        print row


0 0.1 0.3 0.4 0
0 0.2 0.5 0.3 0
0 0.7 0.9 0.2 0

如何从该内存结构中准确删除列（为0）。 如果没有重写和重新读取到另一个临时csv文件来实现这一点，那就更好了。

Answer 1

读取所有行（将所有值映射为浮点数），使用zip(*rows)转换为列，使用any()仅保留具有非零值的列，使用zip(*columns)转换回行：

with open('data.txt', 'rb') as f:
    rows = list(map(float, row) for row in csv.reader(f, delimiter=','))

rows = zip(*[col for col in zip(*rows) if any(col)])

后面的步骤作为演示：

>>> rows = [[0, 0.1, 0.3, 0.4, 0], [0, 0.2, 0.5, 0.3, 0], [0, 0.7, 0.9, 0.2, 0]]
>>> zip(*[col for col in zip(*rows) if any(col)])
[(0.1, 0.3, 0.4), (0.2, 0.5, 0.3), (0.7, 0.9, 0.2)]