檢查給定鍵是否已存在於字典中並將其遞增

Question

如何確定字典中的鍵是否已設置為非無值？

如果那里已經有一個值，我想增加該值，否則將其設置為 1：

my_dict = {}

if my_dict[key] is not None:
  my_dict[key] = 1
else:
  my_dict[key] += 1

Answer 1

您正在尋找collections.defaultdict （適用於 Python 2.5+）。 這個

from collections import defaultdict

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

會做你想做的。

對於常規的 Python dict ，如果給定鍵沒有值，則在訪問 dict 時不會得到None —— 將引發KeyError 。 因此，如果您想使用常規dict ，而不是您的代碼，您將使用

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

Answer 2

我更喜歡在一行代碼中做到這一點。

my_dict = {}

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

字典有一個函數，get，它有兩個參數——你想要的鍵和一個默認值（如果它不存在）。 我更喜歡這種方法 defaultdict 因為您只想處理這一行代碼中不存在密鑰的情況，而不是無處不在。

Answer 3

我個人喜歡使用setdefault()

my_dict = {}

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

Answer 4

為此，您需要key in dict 。

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

但是，您可能應該考慮使用defaultdict （如 dF 建議的那樣）。

Answer 5

要回答“如何確定該字典中的給定索引是否已設置為非無值”的問題，我更喜歡這個：

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

這符合已經調用的 EAFP 概念（更容易請求寬恕而不是許可）。 它還避免了字典中的重復鍵查找，就像key in my_dict and my_dict[key] is not None如果查找成本很高，這很有趣。

對於您提出的實際問題，即如果存在 int 則遞增，否則將其設置為默認值，我還建議

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

正如安德魯威爾金森的回答一樣。

如果您在字典中存儲可修改的對象，還有第三種解決方案。 一個常見的示例是multimap ，您可以在其中存儲鍵的元素列表。 在這種情況下，您可以使用：

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

如果字典中不存在 key 的值，setdefault 方法會將其設置為 setdefault 的第二個參數。 它的行為就像一個標准的 my_dict[key]，返回鍵的值（可能是新設置的值）。

Answer 6

同意cgoldberg。 我是怎么做的：

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

所以要么像上面那樣做，要么像其他人建議的那樣使用默認字典。 不要使用 if 語句。 那不是Pythonic。

Answer 7

從許多答案中可以看出，有幾種解決方案。 尚未提及 LBYL 的一個實例（在你跳躍之前查看），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

Answer 8

有點晚了，但這應該可以。

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

Answer 9

您嘗試執行此操作的方式稱為 LBYL（在您跳躍之前查看），因為您在嘗試增加您的值之前檢查條件。

另一種方法稱為 EAFP（更容易請求寬恕而不是許可）。 在這種情況下，您只需嘗試該操作（增加值）。 如果失敗，您將捕獲異常並將值設置為 1。這是一種稍微更 Pythonic 的方法（IMO）。

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

Answer 10

這不是直接回答問題，但對我來說，看起來您可能需要collections.Counter的功能。

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))

這導致輸出

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

Answer 11

這是我最近提出的用於解決此問題的單線。 它基於setdefault字典方法：

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

Answer 12

我一直在尋找它，但在網上沒有找到它，然后用 Try/Error 試試運氣並找到了它

my_dict = {}

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