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

时间:2009-01-23 14:35:39

标签: python dictionary

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

即,我想这样做:

my_dict = {}

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

即,如果已存在值,我想增加该值,否则将其设置为1。

12 个答案:

答案 0 :(得分:297)

您正在寻找collections.defaultdict(适用于Python 2.5+)。此

from collections import defaultdict

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

会做你想做的事。

对于常规Python dict,如果给定键没有值,则在访问dict时获取None - {{1}将被提出。因此,如果您想使用常规KeyError,而不是代码,则可以使用

dict

答案 1 :(得分:267)

我更喜欢在一行代码中执行此操作。

my_dict = {}

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

字典有一个函数get,它接受两个参数 - 你想要的密钥,以及一个默认值(如果它不存在)。我更喜欢这种方法defaultdict,因为你只想处理这一行代码中不存在密钥的情况,而不是处处。

答案 2 :(得分:49)

我个人喜欢使用setdefault()

my_dict = {}

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

答案 3 :(得分:48)

你需要key in dict成语。

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

但是,您应该考虑使用defaultdict(建议使用dF)。

答案 4 :(得分:17)

要回答问题“如何判断该字典中的某个索引是否已设置为非无效值”,我更愿意这样做:

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],返回键的值(可能是新设置的值)。

答案 5 :(得分:13)

同意cgoldberg。我是怎么做到的:

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

所以要么像上面那样做,要么像其他人建议的那样使用默认字典。不要使用if语句。那不是Pythonic。

答案 6 :(得分:11)

从许多答案中可以看出,有几种解决方案。 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

答案 7 :(得分:7)

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

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

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

答案 8 :(得分:5)

有点晚但这应该有用。

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

答案 9 :(得分:5)

这不是直接回答这个问题,但对我而言,您似乎可能需要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

答案 10 :(得分:3)

这是我最近提出的解决这个问题的单线程。它基于setdefault字典方法:

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

答案 11 :(得分:1)

我一直在寻找它,没有在网上找到它然后尝试了我的运气与Try / Error并找到它

my_dict = {}

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