python当键存在时向值添加元素，键不存在时则新建的简洁方法

在 Python 中，可以使用字典来存储集合对象，并且在键存在时向集合中添加元素，键不存在时则新建一个集合并添加元素。可以使用

dict.setdefault()

方法来简化这个过程，或者手动检查键是否存在。

下面是一些示例代码：

使用

setdefault()

方法

setdefault()

方法会检查字典中是否存在指定键，如果不存在则设置键的默认值（如空集合），然后返回该键对应的值。

# 创建一个空字典
my_dict ={}# 添加元素到集合
key ='fruits'
element ='apple'# 使用 setdefault 方法
my_dict.setdefault(key,set()).add(element)# 打印字典print(my_dict)# 输出: {'fruits': {'apple'}}

使用

defaultdict

类

defaultdict

是

collections

模块中的一个类，它在键不存在时会自动创建默认值。

from collections import defaultdict

# 创建一个默认值为集合的 defaultdict
my_dict = defaultdict(set)# 添加元素到集合
key ='fruits'
element ='apple'# 直接添加元素
my_dict[key].add(element)# 打印字典print(my_dict)# 输出: defaultdict(<class 'set'>, {'fruits': {'apple'}})

手动检查键是否存在

可以手动检查键是否存在于字典中，然后根据情况添加元素或新建集合。

# 创建一个空字典
my_dict ={}# 添加元素到集合
key ='fruits'
element ='apple'# 手动检查键是否存在if key notin my_dict:
    my_dict[key]=set()
my_dict[key].add(element)# 打印字典print(my_dict)# 输出: {'fruits': {'apple'}}

完整示例

以下示例展示了如何向字典中添加多个键和元素，同时处理键存在和不存在的情况：

# 创建一个初始字典
my_dict ={'initial_key':{'initial_value'}}# 定义键和值
keys_and_elements =[('fruits','apple'),('fruits','banana'),('vegetables','carrot'),('fruits','orange'),('vegetables','broccoli'),('initial_key','another_value')]# 使用 setdefault 方法for key, element in keys_and_elements:
    my_dict.setdefault(key,set()).add(element)# 打印最终字典print(my_dict)# 输出:# {'initial_key': {'initial_value', 'another_value'},#  'fruits': {'apple', 'banana', 'orange'},#  'vegetables': {'carrot', 'broccoli'}}