Как разбить список слов по категориям?

Question

[Артем]

день добрый всем! решаю одну задачку - делаю вариацию на игру "угадай слово" (человек указывает какой длины должно быть слово, комп формирует список слов заданной длины из всего массива слов в txt файле и выбирает рандомно слово из этого списка; человек угадывает по буквам это слово за N-е количество раз). Застопорился на одной проблеме. В задании есть такой пункт:
- после того, как пользователь вводит предполагаемую букву, вам нужно разбить слова из списка на категории.
Категории формируются вот по такому принципу:

т.е. если есть на входе есть список, например если загадано слово из 4 букв и из всего списка слов вот эти слова имеют длину в 4 буквы: ALLY BETA COOL DEAL ELSE FLEW GOOD HOPE IBEX и пользователь загадал букву E, то на выходе мне нужно получить словарь:
{"----": ["ALLY", "COOL" , "GOOD"],
"-E--": ["BETA", "DEAL"],
"--E-": ["FLEW", "IBEX"],
"E--E": ["ELSE"],
"---E": ["HOPE"],
}
Может кто подскажет алгоритм, как получать такой словарь? Причем длина слова может варьироваться от 2 до 8 букв...

Comments

[javedimka]

Ну, например, так:

from collections import defaultdict

words_list = ['ALLY', 'BETA', 'COOL', 'DEAL', 'ELSE', 'FLEW', 'GOOD', 'HOPE', 'IBEX']

def make_primary_dict(user_input, words_list):
    guess_dict = defaultdict(list)
    for word in words_list:
        new_word = []
        for ch in word:
            if ch != user_input:
                ch = '-'
                new_word.append(ch)
            else:
                new_word.append(ch)
        guess_dict[(''.join(new_word))] += [word]
    return dict(guess_dict)

>>> make_primary_dict('E', words_list)
{'----': ['ALLY', 'COOL', 'GOOD'], '-E--': ['BETA', 'DEAL'], 'E--E': ['ELSE'], '--E-': ['FLEW', 'IBEX'], '---E': ['HOPE']}

Или упростить, по примеру решения автора:

def make_primary_dict(user_input, words_list):
    guess_dict = defaultdict(list)
    for word in words_list:
        guess_dict[''.join(c if c == user_input else "-" for c in word)] += [word]
    return dict(guess_dict)

>>> make_primary_dict('E', words_list)
{'----': ['ALLY', 'COOL', 'GOOD'], '-E--': ['BETA', 'DEAL'], 'E--E': ['ELSE'], '--E-': ['FLEW', 'IBEX'], '---E': ['HOPE']}

Только толку?

Answer 1

[Артем]

сам нашел решение. может кривовато написано, но зато работает :) :

word_list = ["ALLY", "BETA", "COOL", "DEAL", "ELSE", "FLEW", "GOOD", "HOPE", "IBEX"]
guess = "E"

keys = []

def get_template(word, guess):
    template = "".join(c if c in guess else "-" for c in word)
    return template

for word in word_list:
    template = "".join(c if c in guess else "-" for c in word)
    if template not in keys:
        keys.append(template)
categories = {}.fromkeys([key for key in keys])
for key in categories:
    categories[key] = []
    for word in word_list:
        if get_template(word, guess) == key:
            categories[key].append(word)

print(categories)

Comments

[javedimka]

Яж написал в комментах, а если уж выражения-генераторы знаешь то и тот пример можно укоротить на 7 строк.