нахождение частот парных элементов в списке пар

Question

Допустим, у меня есть длинный список этого типа:

text = [ ['a', 'b'], ['a', 'd'], ['w', 'a'], ['a', 'b'], ... ] 

Учитывая первые элементы, я хочу, чтобы построить словарь, который будет показывать счетчик второй элементы. Например, в конкретном примере выше, я хотел бы иметь что-то вроде этого:

{'a': {'b':2, 'd':1}, 
'w': {'a':1} 
} 

Вот как я безуспешно пытался решить. Я построил список уникальных первых элементов. Давайте назовем это words, а затем:

dic = {} 

for word in words: 
    inner_dic = {} 
    for pair in text: 
    if pair[0] == word: 
     num = text.count(pair) 
     inner_dic[pair[1]] = num 
    dic[pair[0]] = inner_dic 

я получаю явно ошибочный результат. Одна из проблем с кодом заключается в том, что он переопределяет пары. Я не уверен, как это решить.

Answer 1

Вы должны сделать это вместо того, чтобы:

for word in words: 
    inner_dic = {} 
    for pair in text: 
    if pair[0] == word: 
     num = text.count(pair) 
     inner_dic[pair[1]] = num 
    dic[word] = inner_dic 

то есть, вы должны делать dic[word], а не dic[pair[0]], который назначит inner_dic к первому элементу в последнем pair проверено, даже если pair[0] не word.

Answer 2

Вы можете использовать defaultdict в сочетании с Counter Dict:

from collections import Counter, defaultdict 
d = defaultdict(Counter) 

text = [ ['a', 'b'], ['a', 'd'], ['w', 'a'], ['a', 'b'] ] 

for k, v in text: 
    d[k][v] += 1 # for single value 
    # d[k].update(v) for multiple values i.e list of words 

from pprint import pprint as pp 

pp(d) 
{'a': Counter({'b': 2, 'd': 1}), 
'w': Counter({'a': 1})} 

defaultdict создаст новый спаривание ключ/значение, где значение является счетчиком ДИКТ, если ключ не существует, если ключ существует мы просто обновляем значение с помощью параметра Counter.update, который, в отличие от dict.update, будет увеличивать значение, не переписываемое.

используя обычный dict без импорта мы можем использовать dict.setdefault, который будет создавать новую Dict в качестве значения для каждого ключа k и установить значение по умолчанию 0 для каждого подраздела v:

d = {} 
text = [ ['a', 'b'], ['a', 'd'], ['w', 'a'], ['a', 'b'] ] 

for k, v in text: 
    d.setdefault(k, {}).setdefault(v,0) 
    d[k][v] += 1 
pp(d) 
{'a': {'b': 2, 'd': 1}, 'w': {'a': 1}} 

Answer 3

collections module делает коротким работа таких задач.

Используйте счетчик Counter (это словарь, который возвращает 0 для отсутствующих значений, что упрощает использование +=1 для увеличения количества очков). Используйте defaultdict для внешнего Dict (она может автоматически создать новый счетчик для каждого «первого» префикса):

>>> from collections import defaultdict, Counter 
>>> d = defaultdict(Counter) 
>>> text = [ ['a', 'b'], ['a', 'd'], ['w', 'a'], ['a', 'b']] 
>>> for first, second in text: 
    d[first][second] += 1 

Вот эквивалент, используя обычные словари:

text = [ ['a', 'b'], ['a', 'd'], ['w', 'a'], ['a', 'b']] 

d = {} 
for first, second in text: 
    if first not in d: 
     d[first] = {} 
    inner_dict = d[first] 
    if second not in inner_dict: 
     inner_dict[second] = 0 
    inner_dict[second] += 1 

Либо короткий путь или длинный путь будет отлично работать с json module (оба Счетчик и defaultdict - это виды диктов, которые могут быть закодированы JSON).

Надеюсь, это поможет. Удачи вам в вашем анализе текста :-)

Answer 4

text = [ ['a', 'b'], ['a', 'd'], ['w', 'a'], ['a', 'b']] 
d = {} 
for i in text: 
    if d.get(i[0]): 
     if d[i[0]].get(i[1]): 
      d[i[0]][i[1]] +=1 
     else: 
      d[i[0]][i[1]] = 1 
    else: 
     d[i[0]] = {i[1] : 1} 
print d 
>>>{'a': {'b': 2, 'd': 1}, 'w': {'a': 1}} 

Answer 5

Вот способ, используя метод .setdefault:

text = [ ['a', 'b'], ['a', 'd'], ['w', 'a'], ['a', 'b'] ] 
result={} 
for x, y in text: 
    result.setdefault(x, {}).setdefault(y,0) 
    result[x][y]+=1 

>>> result 
{'a': {'b': 2, 'd': 1}, 'w': {'a': 1}} 

Нет внешних библиотек требуется.