Kiểm tra xem một khóa đã cho có tồn tại trong từ điển không và tăng nó

Đưa ra một từ điển, làm thế nào tôi có thể tìm ra nếu một khóa nhất định trong từ điển đó đã được đặt thành giá trị không phải là Không?

Tức là tôi muốn làm điều này:

my_dict = {}

if (my_dict[key] != None):
  my_dict[key] = 1
else:
  my_dict[key] += 1

Tức là tôi muốn tăng giá trị nếu đã có một cái ở đó hoặc đặt nó thành 1 nếu không.

Bạn đang tìm kiếm collections.defaultdict(có sẵn cho Python 2.5+). Điều này

from collections import defaultdict

my_dict = defaultdict(int)
my_dict[key] += 1

sẽ làm những gì bạn muốn.

Đối với các Python thông thường dict, nếu không có giá trị cho một khóa nhất định, bạn sẽ không nhận được Nonekhi truy cập dict - a KeyErrorsẽ được nâng lên. Vì vậy, nếu bạn muốn sử dụng thường xuyên dict, thay vì mã của bạn, bạn sẽ sử dụng

if key in my_dict:
    my_dict[key] += 1
else:
    my_dict[key] = 1

Tôi thích làm điều này trong một dòng mã.

my_dict = {}

my_dict [some_key] = my_dict.get (some_key, 0) + 1

Từ điển có một hàm, get, có hai tham số - khóa bạn muốn và giá trị mặc định nếu nó không tồn tại. Tôi thích phương thức này hơn defaultdict vì bạn chỉ muốn xử lý trường hợp khóa không tồn tại trong một dòng mã này, không phải ở mọi nơi.

Cá nhân tôi thích sử dụng setdefault()

my_dict = {}

my_dict.setdefault(some_key, 0)
my_dict[some_key] += 1

Bạn cần key in dictthành ngữ cho điều đó.

if key in my_dict and not (my_dict[key] is None):
  # do something
else:
  # do something else

Tuy nhiên, có lẽ bạn nên cân nhắc sử dụng defaultdict(như dF đề xuất).

Để trả lời câu hỏi " làm thế nào tôi có thể tìm ra nếu một chỉ mục nhất định trong chính tả đó đã được đặt thành giá trị không phải là Không ", tôi thích điều này:

try:
  nonNone = my_dict[key] is not None
except KeyError:
  nonNone = False

Điều này phù hợp với khái niệm EAFP đã được viện dẫn (dễ dàng hơn để yêu cầu sự tha thứ sau đó cho phép). Nó cũng tránh việc tra cứu khóa trùng lặp trong từ điển như trongkey in my_dict and my_dict[key] is not None rất thú vị nếu tra cứu đắt tiền.

Đối với vấn đề thực tế mà bạn đã đặt ra, tức là tăng int nếu nó tồn tại hoặc đặt nó thành giá trị mặc định, tôi cũng khuyên bạn nên

my_dict[key] = my_dict.get(key, default) + 1

như trong câu trả lời của Andrew Wilkinson.

Có một giải pháp thứ ba nếu bạn đang lưu trữ các đối tượng có thể sửa đổi trong từ điển của bạn. Một ví dụ phổ biến cho điều này là một multimap , nơi bạn lưu trữ một danh sách các yếu tố cho các khóa của bạn. Trong trường hợp đó, bạn có thể sử dụng:

my_dict.setdefault(key, []).append(item)

Nếu một giá trị cho khóa không tồn tại trong từ điển, phương thức setdefault sẽ đặt nó thành tham số thứ hai của setdefault. Nó hoạt động giống như một my_dict [key] tiêu chuẩn, trả về giá trị cho khóa (có thể là giá trị mới được đặt).

Đồng ý với cgoldberg. Làm thế nào tôi làm điều đó là:

try:
    dict[key] += 1
except KeyError:
    dict[key] = 1

Vì vậy, hoặc làm như trên, hoặc sử dụng một dict mặc định như những người khác đã đề xuất. Đừng sử dụng nếu báo cáo. Đó không phải là Pythonic.

Như bạn có thể thấy từ nhiều câu trả lời, có một số giải pháp. Một trường hợp của LBYL (xem trước khi bạn nhảy) chưa được đề cập, phương thức has_key ():

my_dict = {}

def add (key):
    if my_dict.has_key(key):
        my_dict[key] += 1
    else:
        my_dict[key] = 1

if __name__ == '__main__':
    add("foo")
    add("bar")
    add("foo")
    print my_dict

Cách bạn đang cố gắng thực hiện được gọi là LBYL (xem trước khi bạn nhảy), vì bạn đang kiểm tra các điều kiện trước khi cố gắng tăng giá trị của mình.

Cách tiếp cận khác được gọi là EAFP (dễ dàng hơn để xin sự tha thứ sau đó cho phép). Trong trường hợp đó, bạn chỉ cần thử thao tác (tăng giá trị). Nếu thất bại, bạn bắt ngoại lệ và đặt giá trị thành 1. Đây là cách Pythonic hơn một chút để thực hiện (IMO).

http://mail.python.org/pipermail/python-list/2003-May/205182.html

Một chút muộn nhưng điều này sẽ làm việc.

my_dict = {}
my_dict[key] = my_dict[key] + 1 if key in my_dict else 1

Điều này không trực tiếp trả lời câu hỏi, nhưng với tôi, có vẻ như bạn có thể muốn chức năng của các bộ sưu tập .

from collections import Counter

to_count = ["foo", "foo", "bar", "baz", "foo", "bar"]

count = Counter(to_count)

print(count)

print("acts just like the desired dictionary:")
print("bar occurs {} times".format(count["bar"]))

print("any item that does not occur in the list is set to 0:")
print("dog occurs {} times".format(count["dog"]))

print("can iterate over items from most frequent to least:")
for item, times in count.most_common():
    print("{} occurs {} times".format(item, times))

Điều này dẫn đến kết quả đầu ra

Counter({'foo': 3, 'bar': 2, 'baz': 1})
acts just like the desired dictionary:
bar occurs 2 times
any item that does not occur in the list is set to 0:
dog occurs 0 times
can iterate over items from most frequent to least:
foo occurs 3 times
bar occurs 2 times
baz occurs 1 times

Đây là một lớp lót mà tôi đã đưa ra gần đây để giải quyết vấn đề này. Nó dựa trên phương thức từ điển setdefault :

my_dict = {}
my_dict[key] = my_dict.setdefault(key, 0) + 1

Tôi đang tìm kiếm nó, không tìm thấy nó trên web sau đó đã thử vận ​​may của mình với Thử / Lỗi và tìm thấy nó

my_dict = {}

if my_dict.__contains__(some_key):
  my_dict[some_key] += 1
else:
  my_dict[some_key] = 1