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()保留任何具有非零值的列，使用{{转换回行1}}：

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)])

删除python中的列

1 个答案: