Python：将数据写入输出文件的多列

Question

我正在编写一个 python 解决方案来读取名为 Input.txt (tsv) 的文件的第一列并写入名为 Output.txt (tsv) 的文件。

    import csv

    with open('Input.txt', 'r') as rf:
        reader = csv.reader(rf, delimiter='\t')

        with open('Output.txt', 'w') as wf:
            writer = csv.writer(wf, delimiter='\t')

            desired_column = [0]

            for line in reader:
                Column1 = list(line[i] for i in desired_column)
                writer.writerow(Column1)

以上代码读取Input.txt的第一列并写入Output.txt的第一列。

我的问题是：如果我想把同样的数据也写入Output.txt

的第二列怎么办

Answer 1

import csv

with open('Input.txt', 'r') as rf:
    reader = csv.reader(rf, delimiter='\t')

    with open('Output.txt', 'w') as wf:
        writer = csv.writer(wf, delimiter='\t')

        desired_column = [0]
        for line in reader:
            Column1 = list(line[i] for i in desired_column)
            '''We can use writerows function to zip the value. Instead of
               different column name , we can use same copied column to satisfy your requirement'''   
            writer.writerows(zip(Column1 ,Column1 ))

希望这能满足您的要求。查看此内容https://docs.python.org/2/library/csv.html#csv.csvwriter.writerows

Python：将数据写入输出文件的多列

Python: Write the data into multiple columns of output file

csv

python-3.5

read-write