Итерация по словарю Python и специальное добавление в новый список? - PullRequest
0 голосов
/ 21 февраля 2019

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

Например: ввод: {'A':1, 'B':2}.Ожидаемый результат: ['A', 'B', 'B']

То, что я делаю, не работает.Что я пишу в своей функции, чтобы сделать это?

def get_freq_dict():
    freq_dict = {'J' : 1, 'K' : 1, 'Q' : 1, 'X' : 1, 'Z' : 1,\
                'B' : 2, 'C' : 2, 'F' : 2, 'H' : 2, 'M' : 2, 'P' : 2,\
                'V' : 2, 'W' : 2, 'Y' : 2, '' : 2,\
                'G' : 3, 'D' : 4, 'L' : 4, 'S' : 4, 'U' : 4,\
                'N' : 6, 'R' : 6, 'T' : 6, 'O' : 8, 'A' : 9, 'I' : 9,\
                'E' : 12}
    return freq_dict


def bag_of_letters(freq_dict):
    freq_lst = [] 
    for key, value in freq_dict.items():
        for range in(value):
            freq_lst.append(value)
    return freq_lst


def main():

    freq_dict = get_freq_dict()
    freq_lst = bag_of_letters(freq_dict)

    print(freq_dict, freq_lst)
main()

Ответы [ 5 ]

0 голосов
/ 21 февраля 2019

Вы можете легко сделать это с помощью Counter:

from collections import Counter

d = {'C': 3, 'B': 2, 'A': 1}
c = Counter(d)
list(c.elements())
# ['C', 'C', 'C', 'B', 'B', 'A']
0 голосов
/ 21 февраля 2019

Вы можете использовать тот факт, что

Если items (), keys (), values ​​(), iteritems (), iterkeys () и itervalues ​​() вызываются без промежуточных модификацийсловарь, списки будут напрямую соответствовать.

, как показано в этой теме , поэтому вы можете просто сделать:

import itertools
x = {'A':1, 'B':2}
out = list(map(lambda x,y: [x]*y, x.keys(), x.values()))
print(out) #[['A'], ['B', 'B']]
out = list(itertools.chain.from_iterable(out))
print(out) #['A', 'B', 'B']

Я использовал itertools дляСвести список, если вы не хотите import itertools, вы можете сделать:

out = sum(out,[])

вместо

out = list(itertools.chain.from_iterable(out))
0 голосов
/ 21 февраля 2019

Виновник :

for range in(value):
    freq_lst.append(value)

Спасатель :

for i in range(value):
     freq_lst.append(key)

Следовательно :

def get_freq_dict():
    freq_dict = {'J' : 1, 'K' : 1, 'Q' : 1, 'X' : 1, 'Z' : 1,\
                'B' : 2, 'C' : 2, 'F' : 2, 'H' : 2, 'M' : 2, 'P' : 2,\
                'V' : 2, 'W' : 2, 'Y' : 2, '' : 2,\
                'G' : 3, 'D' : 4, 'L' : 4, 'S' : 4, 'U' : 4,\
                'N' : 6, 'R' : 6, 'T' : 6, 'O' : 8, 'A' : 9, 'I' : 9,\
                'E' : 12}
    return freq_dict


def bag_of_letters(freq_dict):
    freq_lst = []
    for key, value in freq_dict.items():
       # print(key, value)
        for i in range(value):
            freq_lst.append(key)
    return freq_lst


def main():

    freq_dict = get_freq_dict()
    freq_lst = bag_of_letters(freq_dict)

    print(freq_lst)
main()

ВЫХОД :

['J', 'K', 'Q', 'X', 'Z', 'B', 'B', 'C', 'C', 'F', 'F', 'H', 'H', 'M', 'M', 'P', 'P', 'V', 'V', 'W', 'W', 'Y', 'Y', '', '', 'G', 'G', 'G', 'D', 'D', 'D', 'D', 'L', 'L', 'L', 'L', 'S', 'S', 'S', 'S', 'U', 'U', 'U', 'U', 'N', 'N', 'N', 'N', 'N', 'N', 'R', 'R', 'R', 'R', 'R', 'R', 'T', 'T', 'T', 'T', 'T', 'T', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'I', 'I', 'I', 'I', 'I', 'I', 'I', 'I', 'I', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E']

ИЛИ

, если вы хотите их красиво спаренных :

 for i in range(value):
     freq_lst.append([key]*value)

OP : Однако у меня все еще есть проблемы с печатным выходом.Это дает мне то, что я ищу, а также оригинальный словарь сверху

Ans : потому что вы печатаете как dict, так и list:

print(freq_dict, freq_lst)

Вместо этого просто напечатайте list:

print(freq_lst)

РЕДАКТИРОВАТЬ 2:

Еще один более приятный способ группировки похожих элементов,используя groupby():

Добавьте только key:

 for i in range(0, value):
      freq_lst.append(key)

и затем:

 print([list(j) for i, j in groupby(freq_lst)])

ВЫХОД :

[['J'], ['K'], ['Q'], ['X'], ['Z'], ['B', 'B'], ['C', 'C'], ['F', 'F'], ['H', 'H'], ['M', 'M'], 
 ['P', 'P'], ['V', 'V'], ['W', 'W'], ['Y', 'Y'], ['', ''], ['G', 'G', 'G'], 
 ['D', 'D', 'D', 'D'], ['L', 'L', 'L', 'L'], ['S', 'S', 'S', 'S'], ['U', 'U', 'U', 'U'], 
 ['N', 'N', 'N', 'N', 'N', 'N'], ['R', 'R', 'R', 'R', 'R', 'R'], 
 ['T', 'T', 'T', 'T', 'T', 'T'], ['O', 'O', 'O', 'O', 'O', 'O', 'O', 'O'], 
 ['A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A'], 
 ['I', 'I', 'I', 'I', 'I', 'I', 'I', 'I', 'I'], 
 ['E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E', 'E']]
0 голосов
/ 21 февраля 2019

Вот вы

freq_dict = {'J' : 1, 'K' : 1, 'Q' : 1, 'X' : 1, 'Z' : 1,\
            'B' : 2, 'C' : 2, 'F' : 2, 'H' : 2, 'M' : 2, 'P' : 2,\
            'V' : 2, 'W' : 2, 'Y' : 2, '' : 2,\
            'G' : 3, 'D' : 4, 'L' : 4, 'S' : 4, 'U' : 4,\
            'N' : 6, 'R' : 6, 'T' : 6, 'O' : 8, 'A' : 9, 'I' : 9,\
            'E' : 12}


def bag_of_letters():
    freq_lst = []

    for key in freq_dict:
        for i in range(0, freq_dict[key]):
            freq_lst.append(key)

    return freq_lst


def main():
    freq_lst = bag_of_letters()
    print(freq_lst)


main()

результат ['J', 'K', 'Q', 'X', 'Z', 'B', 'B', «C», «C», «F», «F», «H», «H», «M», «M», «P», «P», «V», «V», «W ',' W ', ecc]

0 голосов
/ 21 февраля 2019

Вместо этого следует выполнить итерацию range и добавить key вместо value:

for key, value in freq_dict.items():
    for _ in range(value):
        freq_lst.append(key)

В качестве альтернативы вы можете использовать понимание списка:

def bag_of_letters(freq_dict):
    return [i for k, v in freq_dict.items() for i in [k] * v]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...