Как сделать так, чтобы программа отработала многострочный текст?

Question

[n293]

Есть задача:

Во входном файле (вы можете читать данные из файла input.txt) записан текст. Словом считается последовательность непробельных подряд идущих символов. Слова разделены одним или большим числом пробелов или символами конца строки. Для каждого слова из этого текста подсчитайте, сколько раз оно встречалось в этом тексте ранее.

Есть вот такое решение:

A = {}
a = open('input.txt', 'r')
for x in a:
    A[x] = A.get(x, 0) + 1
    print(A[x] - 1, end=' ')

А есть тест, который это решение не проходит:

Входные данные:
She sells sea shells on the sea shore;
The shells that she sells are sea shells I'm sure.
So if she sells sea shells on the sea shore,
I'm sure that the shells are sea shore shells.

Вывод программы:
0 0 0 0 0 0 1 0 0 1 0 0 1 0 2 2 0 0 0 0 1 2 3 3 1 1 4 0 1 0 1 2 4 1 5 0 0

Собственно, как сделать так, чтобы многострочный текст тоже отработал в программе?

Comments

[longclaps]

Вот любопытно, вы отдаёте себе отчет в том, что именно считаете?
Судя по коду - нет.
Может, попробуете?

[n293]

да, я не понимаю. потому и обращаюсь за помощью.

[longclaps]

n293, мне нетрудно объяснить, но что в этом толку?

что именно считаете?

Вот то, что вы считаете (неважно как, но - считаете), вы используете в качестве ключа в словаре.
Нет ли у вас желания поглядеть, что оно такое?

Answer 1

[n293]

Нагуглил другие задачки, в общем. Вытащил из файла текст, запихнул в строку, засплитил и получилось норм.

strWords = ''
counter = {}
f = open('input.txt', 'r')
for line in f:
    strWords = strWords + ' ' + line.replace('\n', '')
f.close()
for word in strWords.split():
    counter[word] = counter.get(word, 0) + 1
    print(counter[word] - 1, end=' ')

Comments

[Andrey Dugin]

1) А в чём смысл сначала соединять слова через пробел, а потом их же сплитить?
2) Зачем читаете файл в память целиком? А если он размером несколько гигабайт?
3) Зачем к значению сначала прибавляете единицу, а потом вычитаете?
4) Плохая практика работать с файлами вне контекста with, т.к. в случае исключения close() не сработает, а с контекстом явное close() не требуется (будет выполнено автоматически).

Answer 2

[Andrey Dugin]

from collections import Counter

counter = Counter()
with open(r'z:\test.txt', 'rt') as file:
    for line in file:
        for word in line.strip().split():
            print(counter[word], end=' ')
            counter[word] += 1

Comments

[n293]

спасибо!