Python. Прочитайте два разных поля текстового файла в двух разных списках со списком.

Question

У меня есть текстовый файл (tab delimeted), и я хотел бы передать поле 1 в списке (list1) и поле 3 в другом списке (list2), используя понимание списка.

Код я написал это:

with open (path) as file1: 
    list1 = [line.strip().split('\t')[0] for line in file1]; 
    list2 = [line.strip().split('\t')[3] for line in file1]; 

Проблема заключается в том, что песни2 остается пустым.

Любые мысли, почему это происходит?

Спасибо!

Answer 1

Вам нужно сбросить pointet обратно в начало файла:

with open (path) as file1: 
    list1 = [line.strip().split('\t')[0] for line in file1] 
    file1.seek(0) 
    list2 = [line.strip().split('\t')[3] for line in file1]; 

Но вы можете сделать это в один проход, просто добавьте к каждому, как вы итерацию так что вам нужно всего лишь перейти на файл содержание раз:

with open (path) as file1: 
    list1, list2 = [], [] 
    for line in file1: 
     spl =line.strip().split('\t') 
     list1.append(spl[0]) 
     list2.append(spl[3]) 

Вы также можете использовать CSV LIB для анализа данных для вас:

import csv 

with open (path) as file1: 
    list1, list2 = [], [] 
    for row in csv.reader(file1, delimiter="\t): 
     list1.append(row[0]) 
     list2.append(row[3])