检查给定键是否已存在于字典中并递增

Question

给定一个字典,如何确定该字典中的给定键是否已设置为非None值？

即,我想这样做:

my_dict = {}

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

即,我想增加值,如果已经存在,或者将其设置为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将引发a.因此,如果您想使用常规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

要回答" 如何判断该字典中的给定索引是否已设置为非None值 "的问题,我更喜欢这样:

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

就像Andrew Wilkinson的回答一样.

如果要在字典中存储可修改对象,则有第三种解决方案.一个常见的例子是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

您尝试这样做的方式称为LBYL(在跳跃之前查看),因为您在尝试增加值之前检查条件.

另一种方法称为EAFP(更容易请求宽恕然后许可).在这种情况下,您只需尝试操作(增加值).如果失败,则捕获异常并将值设置为1.这是一种稍微更加Pythonic的方法(IMO).

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

Answer 9

有点晚但这应该有效.

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

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