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 not in 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'}}