Создайте сводный текстовый файл папки с именем файла и первыми двумя строками из каждого файла

Question

У меня есть каталог с 260+ текстовыми файлами, содержащими информацию о выигрыше. Я хочу создать сводный текстовый файл всех этих файлов, содержащий имя файла и первые две строки каждого файла. Моя идея состояла в том, чтобы создать два списка по отдельности и «сжать» их. Тем не менее, я могу получить список имен файлов, но я не могу получить первые две строки файла в добавленный список. Вот мой код:

# creating a list of filename
for f in os.listdir("../scores"):
    (pdb, extension) = os.path.splitext(f)
    name.append(pdb[1:5])

# creating a list of the first two lines of each file
for f in os.listdir("../scores"):
    for line in open(f):
        score.append(line)
        b = f.nextline()
        score.append(b)

Я получаю сообщение об ошибке: str не имеет атрибута nextline. Пожалуйста, помогите, спасибо заранее.

Answer 1

Проблема, которую вы получаете, является результатом попытки извлечь более одной строки за раз из файла результатов с помощью файлового итератора (for line in f). Вот быстрое решение (один из нескольких способов сделать это, я уверен):

# creating a list of the first two lines of each file
for f in os.listdir("../scores"):
    with open(f) as fh:
        score.append(fh.readline())
        score.append(fh.readline())

Оператор with заботится о закрытии файла после того, как вы закончите, и он предоставляет вам объект дескриптора файла (fh), с которого вы можете захватывать строки вручную.

Answer 2

Файловые объекты имеют метод next(), а не nextline().

Answer 3

Вот моя, возможно, более старомодная версия с перенаправленной печатью для облегчения перевода строки.

## written for Python 2.7, summarize filename and two first lines of files of given filetype
import os
extension = '.txt' ## replace with extension of desired files
os.chdir('.') ## '../scores') ## location of files

summary = open('summary.txt','w')
# creating a list of filenames with right extension
for fn in [fn for fn in os.listdir(os.curdir) if os.path.isfile(fn) and fn.endswith(extension)]:
    with open(fn) as the_file:
        print >>summary, '**'+fn+'**'
        print >>summary, the_file.readline(), the_file.readline(),
        print >>summary, '-'*60
summary.close()
## show resulta
print(open('summary.txt').read())

Answer 4

Объединение комментария от Давида и ответа от perimosocordiae:

from __future__ import with_statement
from itertools import islice
import os
NUM_LINES = 2
with open('../dir_summary.txt','w') as dir_summary:
  for f in os.listdir('.'):
    with open(f) as tf:
      dir_summary.write('%s: %s\n' % (f, repr(', '.join(islice(tf,NUM_LINES)))))