python - 选择一定的值然后输出

Question

我有一个包含混合信息的文件，而我只需要其中的某些列。

下面是我的示例文件。

A B C D       
1 2 3 abcdef   
5 6 7 abcdef  
1 2 3 abcdef  

我想提取文件以获取我需要的信息。例如，在我的输出文件中如下所示。

A C D       # I only need A, C, and D column.
1 3 ab      # For D column, I only need ab.
5 7 ab      
1 3 ab 

它不是 csv 或 txt 文件，而是每列之间有一个空格。

Answer 1

您仍然可以使用分隔符 kwarg 读取带有 csv 模块的空格分隔文件:

>>> with open('/tmp/data.txt') as f:
...    reader = csv.DictReader(f, delimiter=' ')
...    for row in reader:
...        print row['A'], row['C'], row['D'][:2]
...         
1 3 ab
5 7 ab
1 3 ab