defaultdict một bước khởi

Question

Nó sẽ là tiện lợi nếu một defaultdict có thể được khởi tạo dọc theo dòng sau

d = defaultdict(list, (('a', 1), ('b', 2), ('c', 3), ('d', 4), ('a', 2), 
    ('b', 3))) 

để sản xuất

defaultdict(<type 'list'>, {'a': [1, 2], 'c': [3], 'b': [2, 3], 'd': [4]}) 

Thay vào đó, tôi nhận được

defaultdict(<type 'list'>, {'a': 2, 'c': 3, 'b': 3, 'd': 4}) 

Để có được những gì tôi cần, cuối cùng tôi phải làm điều này:

d = defaultdict(list) 
for x, y in (('a', 1), ('b', 2), ('c', 3), ('d', 4), ('a', 2), ('b', 3)): 
    d[x].append(y) 

Đây là bước IMO nhiều hơn cần thiết, tôi có thiếu gì đó ở đây không?

Answer 1

hành vi bạn mô tả sẽ không nhất quán với các hành vi khác của defaultdict.Có vẻ như những gì bạn muốn là FooDict sao cho

>>> f = FooDict() 
>>> f['a'] = 1 
>>> f['a'] = 2 
>>> f['a'] 
[1, 2] 

Chúng tôi có thể làm điều đó, nhưng không phải với defaultdict; cho phép gọi nó là AppendDict

import collections 

class AppendDict(collections.MutableMapping): 
    def __init__(self, container=list, append=None, pairs=()): 
     self.container = collections.defaultdict(container) 
     self.append = append or list.append 
     for key, value in pairs: 
      self[key] = value 

    def __setitem__(self, key, value): 
     self.append(self.container[key], value) 

    def __getitem__(self, key): return self.container[key] 
    def __delitem__(self, key): del self.container[key] 
    def __iter__(self): return iter(self.container) 
    def __len__(self): return len(self.container) 

Answer 2

Sắp xếp và itertools.groupby đi một chặng đường dài:

>>> L = [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('a', 2), ('b', 3)] 
>>> L.sort(key=lambda t:t[0]) 
>>> d = defaultdict(list, [(tup[0], [t[1] for t in tup[1]]) for tup in itertools.groupby(L, key=lambda t: t[0])]) 
>>> d 
defaultdict(<type 'list'>, {'a': [1, 2], 'c': [3], 'b': [2, 3], 'd': [4]}) 

Để làm cho hơn này của một lớp lót:

L = [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('a', 2), ('b', 3)] 
d = defaultdict(list, [(tup[0], [t[1] for t in tup[1]]) for tup in itertools.groupby(sorted(L, key=operator.itemgetter(0)), key=lambda t: t[0])]) 

Hope this helps

Answer 3

gì bạn dường như đang thiếu là rằng defaultdict là một phân lớp đơn giản (không đặc biệt là "huyền diệu") của dict. Tất cả đối số đầu tiên là cung cấp chức năng nhà máy cho thiếu các phím. Khi bạn khởi tạo defaultdict, bạn đang khởi tạo một dict.

Nếu bạn muốn tạo ra

defaultdict(<type 'list'>, {'a': [1, 2], 'c': [3], 'b': [2, 3], 'd': [4]}) 

bạn nên khởi tạo nó theo cách của bạn sẽ khởi tạo bất kỳ khác dict có giá trị là danh sách:

d = defaultdict(list, (('a', [1, 2]), ('b', [2, 3]), ('c', [3]), ('d', [4]))) 

Nếu dữ liệu ban đầu của bạn có được trong dạng tuple có phần tử thứ 2 luôn là số nguyên, sau đó chỉ cần đi với vòng lặp for. Bạn gọi nó thêm một bước; Tôi gọi nó là cách rõ ràng và rõ ràng để làm điều đó.

Answer 4

Tôi nghĩ rằng hầu hết trong số này là rất nhiều khói và gương để tránh một for vòng lặp đơn giản:

di={} 
for k,v in [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('a', 2),('b', 3)]: 
    di.setdefault(k,[]).append(v) 
# di={'a': [1, 2], 'c': [3], 'b': [2, 3], 'd': [4]} 

Nếu mục tiêu của bạn là một trong những dòng và bạn muốn cú pháp lạm dụng mà tôi có thể không hoàn toàn tán thành hoặc hỗ trợ bạn có thể sử dụng một sự hiểu biết tác dụng phụ:

>>> li=[('a', 1), ('b', 2), ('c', 3), ('d', 4), ('a', 2),('b', 3)] 
>>> di={};{di.setdefault(k[0],[]).append(k[1]) for k in li} 
set([None]) 
>>> di 
{'a': [1, 2], 'c': [3], 'b': [2, 3], 'd': [4]} 

Nếu bạn thực sự muốn đi quá nhiệt tình vào không đọc được:

>>> {k1:[e for _,e in v1] for k1,v1 in {k:filter(lambda x: x[0]==k,li) for k,v in li}.items()} 
{'a': [1, 2], 'c': [3], 'b': [2, 3], 'd': [4]} 

Bạn không muốn làm điều đó. Sử dụng vòng lặp Luke!

Answer 5

>>> kvs = [(1,2), (2,3), (1,3)] 
>>> reduce(
... lambda d,(k,v): d[k].append(v) or d, 
... kvs, 
... defaultdict(list)) 
defaultdict(<type 'list'>, {1: [2, 3], 2: [3]})