Разбор файла с разделителями табуляции в отдельные списки или строки
Я пытаюсь взять файл с разделителями табуляции с двумя столбцами "Имя и возраст", который читается следующим образом:
'Имя\Tage\nMark\t32\nMatt\t29\nJohn\T67\nJason\t45\nMatt\t12\nFrank\t11\nFrank\t34\nFrank\T65\nFrank\T78\п'
И просто создайте два списка, один с именами (называемыми именами, без заголовка) и один с возрастом (называемый возрастом, но без возрастов в списке).
Ответы
Ответ 1
Используя модуль csv, вы можете сделать что-то вроде этого:
import csv
names=[]
ages=[]
with open('data.csv','r') as f:
next(f) # skip headings
reader=csv.reader(f,delimiter='\t')
for name,age in reader:
names.append(name)
ages.append(age)
print(names)
# ('Mark', 'Matt', 'John', 'Jason', 'Matt', 'Frank', 'Frank', 'Frank', 'Frank')
print(ages)
# ('32', '29', '67', '45', '12', '11', '34', '65', '78')
Ответ 2
данные с разделителями табуляции находятся в домене модуля csv
:
>>> corpus = 'Name\tAge\nMark\t32\nMatt\t29\nJohn\t67\nJason\t45\nMatt\t12\nFrank\t11\nFrank\t34\nFrank\t65\nFrank\t78\n'
>>> import StringIO
>>> infile = StringIO.StringIO(corpus)
infile
что infile
был просто обычным file
...
>>> import csv
>>> r = csv.DictReader(infile,
... dialect=csv.Sniffer().sniff(infile.read(1000)))
>>> infile.seek(0)
вам даже не нужно сообщать модуль csv о заголовках и формате разделителя, он сам это выяснит
>>> names, ages = [],[]
>>> for row in r:
... names.append(row['Name'])
... ages.append(row['Age'])
...
>>> names
['Mark', 'Matt', 'John', 'Jason', 'Matt', 'Frank', 'Frank', 'Frank', 'Frank']
>>> ages
['32', '29', '67', '45', '12', '11', '34', '65', '78']
>>>
Ответ 3
Я бы использовал методы split
и splitlines
строк:
names = []
ages = []
for name_age in input.splitlines():
name, age = name_age.strip().split("\t")
names.append(name)
ages.append(age)
Если бы вы разбирали более сложный формат, я бы предложил использовать модуль csv, который также может обрабатывать tsv... Но похоже, что это будет немного переборщить здесь.
Ответ 4
Ответ Unutbu сжат, используя понимание списка:
names = [x[0] for x in csv.reader(open(filename,'r'),delimiter='\t')]
ages = [x[1] for x in csv.reader(open(filename,'r'),delimiter='\t')]