如何从python中的txt中选择不同的列

时间:2016-03-14 22:01:59

标签: python

我想分别打印出我的第一列与其他列(作为标签分隔)。例如Col1 + Col2用于第一次打印,Col1 + Col3用于第二Col1 + Col4等。但是我用下面的代码打印出Col1 + Col2。为什么循环没有进展?

import csv
with open('/Users/elf/Desktop/TEST_DATA/text_haplo.txt', 'r', encoding="utf-8", errors="ignore") as text_haplo:
    next(text_haplo)
    for i in range(1, 5):
        for line in text_haplo:
            line = line.split('\t')
            print(line[0], line[i])

输入是:

A    1    2    3
B    10   20   30
C    100  200  300

我希望输出像:

A 1
B 10
C 100

A 2
B 20
C 200

A 3
B 30
C 300

1 个答案:

答案 0 :(得分:1)

更新版本:

for i in range(1, 4):
    with open('test.csv') as f:
        for line in f:
            line = line.rstrip().split('\t')
            print(line[0], line[i])
        print()

但是,一次又一次地读取文件并不是一个好习惯。只有当您的目标机器缺少RAM或文件非常大时才可接受。否则最好将其读取一次到内存然后按所需顺序输出:

data = []
with open('test.csv') as f:
    for line in f: data.append(line.rstrip().split('\t'))

for i in range(1, 4):
    for columns in data:
        print(columns[0], columns[i])
    print()