Как из txt файла вытащить строки с одинаковым началом?

Question

[Elbro]

Задача достаточно простая, но что-то никак не могу в голове обрисовать ее решение на практику.

Задача такая, есть список:

family = ['mom', 'dad', 'brother']

Есть txt файл с диалогом, который начинается с человека

mom - Hello, do you need anything in the shop?
brother - Yes, bring some fruits
dad - And beer
mom - okey
dad - thx
brother - thx

Задача вытащить циклом(?) сообщения каждого.

Как я это вижу:
Считывать весь текст и искать абонента
Затем if state по абоненту, мол мама писала следующие сообщение, отец такие, а брат такие. Что-то никак не могу понять как это сделать в минимум кода.

f=open("test.rtf","r")
for line in f:
    words=line.split()
    if words:
        print(words[0])
        print(line)

f.close()

Пока только нашел как искать первое слово, но как сделать чтобы печатать только содержащее mom, затем dad и тд

Answer 1

[Алексей Черемисин]

f=open("test.rtf","r")
for line in f:
    words=line.split()
    if words:
        if words[0] == "mom":
           print(words[0])
           print(line)

f.close()

Comments

[Elbro]

Во, то, что надо. только почему-то первую строку пропускает, но тут я разберусь!

[Ипатьев]

Elbro, а для братика весь этот код будете повторять ещё раз, серьёзно?

[Алексей Черемисин]

Ипатьев, ну, решение есть, а дальше уж пусть сам думает :)

[Ипатьев]

Действительно. Главное заветную птичку для вопроса получить, а там хоть трава не расти.

[Elbro]

Ипатьев, круче.
c=0
list = [‘mom’, ‘dad’, ‘brother’]
if words[0] == list[c]
…
c+=1

Сорри за кривотню, с телефона пишу, но это будут переменные

Answer 2

[Алан Гибизов]

data = '''mom a
dad b
dad c
mom d
dad e'''
start_text = 'mom'
[print(item) for item in data.splitlines() if item.startswith(start_text)]

Answer 3

[Vindicar]

Очень просто.

dialogue = dict()  # обычный питоновский словарь

# datasource - коллекция строк. Но данные можно читать и построчно из файла 
# with open('text.txt', 'rt') as datasource:
datasource = """mom - Hello, do you need anything in the shop?
brother - Yes, bring some fruits
dad - And beer
mom - okey
dad - thx
brother - thx
""".split()  

for line in datasource:
    actor, _, text = line.split(' - ')  # разбиваем строку на две части
    if _ is not None:  # нашли разделитель, действуем
        # если нужного ключа не было, создать его с указанным значением,
        # (в нашем случае с новым пустым списком) и вернуть это значение
        # а если ключ был, то просто вернуть его значение
        actor_lines = dialogue.setdefault(actor.lower(), [])
        actor_lines.append(text)
print(dialogue)

Можно и с defaultdict, так несколько удобнее:

from collections import defaultdict 
# если у словаря спрашивают неизвестный ключ, он по-тихому 
# поместит по этому ключу пустой список, и отдаст этот список
# этот словарь будет хранить результат
dialogue = defaultdict(list)  # list - фабрика новых значений

# datasource - коллекция строк. Но данные можно читать и построчно из файла 
# with open('text.txt', 'rt') as datasource:
datasource = """mom - Hello, do you need anything in the shop?
brother - Yes, bring some fruits
dad - And beer
mom - okey
dad - thx
brother - thx
""".split()  

for line in datasource:
    actor, _, text = line.split(' - ')  # разбиваем строку на две части
    if _ is not None:  # нашли разделитель, действуем
        # если нужного значения ключа не было, defaultdict сам создаст запрошенный ключ
        # так что мы гарантированно получим список
        dialogue[actor.lower()].append(text)

print(dialogue)

Answer 4

[Сергей П]

Подсказка: отсортировать.

Comments

[Elbro]

Отсортировал, теперь это единый список где идет

мама
мама
мама
папа
папа
папа.

Не подскажите как разбить их в отдельные единицы?

[Алексей Черемисин]

Зачем их сортировать? Если у человека стоит задача совсем другая - отфильтровать!

[Сергей П]

Окей, отсортировать сумел,
Тогда следующая подсказка: split и groupby.
У split есть один полезнй необязательный аргумент.

[Сергей П]

Алексей Черемисин, Где вы в постановке задачи увидели слово "отфильтровать"?
На сколько я понял, автору вопроса нужно сгруппировать реплики по первому слову. Фильтрация - это другое.