Thêm vào danh sách trong từ điển Python

Question

Có cách nào thanh lịch hơn để viết mã này không?

Điều tôi đang làm: Tôi có khóa và ngày. Có thể có một số ngày được chỉ định cho một khóa và vì vậy tôi đang tạo một từ điển danh sách các ngày để biểu diễn điều này. Mã sau hoạt động tốt, nhưng tôi đã hy vọng cho một phương pháp thanh lịch và Pythonic hơn.

dates_dict = dict() 
for key, date in cur: 
    if key in dates_dict: 
     dates_dict[key].append(date) 
    else: 
     dates_dict[key] = [date] 

Tôi đã mong đợi những điều sau đây hoạt động, nhưng tôi tiếp tục nhận được NoneType không có lỗi thêm thuộc tính.

dates_dict = dict() 
for key, date in cur: 
    dates_dict[key] = dates_dict.get(key, []).append(date) 

này có lẽ có cái gì để làm với thực tế rằng

print([].append(1)) 
None 

nhưng tại sao?

Answer 1

list.append lợi nhuận None, vì nó là một hoạt động tại chỗ và bạn đang gán cho nó trở lại dates_dict[key]. Vì vậy, lần sau khi bạn thực hiện dates_dict.get(key, []).append, bạn đang thực sự đang thực hiện None.append. Đó là lý do tại sao nó thất bại. Thay vào đó, bạn có thể chỉ cần thực hiện

dates_dict.setdefault(key, []).append(date) 

Nhưng, chúng tôi chỉ có collections.defaultdict cho mục đích này. Bạn có thể làm điều gì đó như thế này

from collections import defaultdict 
dates_dict = defaultdict(list) 
for key, date in cur: 
    dates_dict[key].append(date) 

Điều này sẽ tạo một đối tượng danh sách mới nếu không tìm thấy key trong từ điển.

Lưu ý: Vì defaultdict sẽ tạo danh sách mới nếu khóa không được tìm thấy trong từ điển, điều này sẽ có tác dụng phụ không chủ ý. Ví dụ, nếu bạn chỉ muốn lấy một giá trị cho khóa, mà không có ở đó, nó sẽ tạo một danh sách mới và trả về nó.

Answer 2

dates_dict[key] = dates_dict.get(key, []).append(date) đặt dates_dict[key] thành None là list.append trả về None.

In [5]: l = [1,2,3] 

In [6]: var = l.append(3) 

In [7]: print var 
None 

Bạn nên sử dụng collections.defaultdict

import collections 
dates_dict = collections.defaultdict(list) 

Answer 3

Có cách nào thanh lịch hơn để viết mã này không?

Sử dụng collections.defaultdict:

from collections import defaultdict 

dates_dict = defaultdict(list) 
for key, date in cur: 
    dates_dict[key].append(date)