Python: запись данных в несколько столбцов выходного файла - PullRequest
Новая
       66

Python: запись данных в несколько столбцов выходного файла

0 голосов
/ 05 июня 2018

Я пишу решение на 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

1 Ответ

0 голосов
/ 11 июня 2018 
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

...