Запись списка фраз в csv файл
Следуя предыдущему посту , я написал некоторый код Python для вычисления частоты появления определенных фраз (содержащихся в переменной «word_list» с тремя перечисленными примерами, но их будет гораздо больше) в большом количестве текстовых файлов. . Код, который я написал ниже, требует от меня взять каждый элемент списка и вставить его в строку для сравнения с каждым текстовым файлом. Однако текущий код записывает частоты только для последней фразы в списке, а не все из них в соответствующие столбцы электронной таблицы. Это просто проблема с отступом, не размещение строки записи в правильном положении или в моем коде есть логическая ошибка. Также есть ли способ избежать использования списка для присвоения строк, чтобы сравнить фразы с фразами в текстовых файлах?
word_list = ['in the event of', 'frankly speaking', 'on the other hand']
S = {}
p = 0
k = 0
with open(file_path, 'w+', newline='') as csv_file:
writer = csv.writer(csv_file)
writer.writerow(["Fohone-K"] + word_list)
for filename in glob.glob(os.path.join(path, '*.txt')):
if filename.endswith('.txt'):
f = open(filename)
Fohone-K = filename[8:]
data = f.read()
# new code section from scratch file
l = len(word_list)
for s in range(l):
phrase = word_list[s]
S = data.count((phrase))
if S:
#k = k + 1
print("'{}' match".format(Fohone-K), S)
else:
print("'{} no match".format(Fohone-K))
print("\n")
# for m in word_list:
if S >= 0:
print([Fohone-K] + [S])
writer.writerow([Fohone-K] + [S])
В настоящее время вывод выглядит следующим образом.
Введите здесь описание изображения
Когда это должно выглядеть так.
Введите здесь описание изображения
Как мне заставить это записывать вывод в виде строки, а не всего в виде столбца?
Можете ли вы предоставить образец входного файла, можете ли вы показать, что вы сейчас получаете на выходе, и что вы хотели бы получить?
Входные данные — это просто набор текстовых файлов в каталоге, каждый из которых содержит большое количество текста. На выходе должен быть файл .csv с именем файла в первом столбце и подсчетом количества фраз в следующей строке. Что я получаю, так это имя текстового файла, повторяющееся в первом столбце (в зависимости от количества фраз в списке) и количество фраз для каждого файла во втором столбце. Я добавил снимки экрана фактического и требуемого вывода выше в исходном вопросе.
Пожалуйста, исправьте свои синтаксические ошибки и отступы перед публикацией кода - опубликованный вами код не будет работать, даже если у кого-то есть входные файлы.
Я только что запустил код на своей машине, и он работает без сообщения об ошибке об отступе.
Ответы 1
Вы, вероятно, собирались что-то вроде этого:
import csv, glob, os
word_list = ['in the event of', 'frankly speaking', 'on the other hand']
file_path = 'out.csv'
path = '.'
with open(file_path, 'w+', newline='') as csv_file:
writer = csv.writer(csv_file)
writer.writerow(["Fohone-K"] + word_list)
for filename in glob.glob(os.path.join(path, '*.txt')):
if filename.endswith('.txt'):
with open(filename) as f:
postfix = filename[8:]
content = f.read()
matches = [content.count(phrase) for phrase in word_list]
print(f"'{filename}' {'no ' if all(n == 0 for n in matches) else ''}match")
writer.writerow([postfix] + matches)
Основная проблема заключалась в том, что вы писали S в каждой строке, которая содержала только один счет. Здесь это исправлено путем написания полного набора совпадений.
Только что попробовал ваш код на небольшом участке фраз и текстовых файлов, и он отлично работает. Завтра проверю на большей выборке, но спасибо за помощь.
Возможно, сейчас самое время попрактиковаться в навыках отладки. Следующие две ссылки дают отличные советы по отладке вашего кода. Шесть методов отладки для программистов на Python или Полное руководство по отладке Python