能否将Pandas示例中的两次replace方法合并为一个？（兼顾扩展性）

当然可以，你只需要把custom_groups转换成值到分组名的反向映射字典，就能用一次replace完成所有替换，后续新增分组时只需修改custom_groups，无需改动replace逻辑，扩展性直接拉满。

实现代码 #

import pandas as pd

custom_groups = {"odd": [1, 3, 5], "even": [2, 4]}

# 构建反向映射字典：{1: 'odd', 3: 'odd', 5: 'odd', 2: 'even', 4: 'even'}
reverse_map = {num: group for group, nums in custom_groups.items() for num in nums}

s = pd.Series([1, 2, 3, 4, 5])
# 一次replace完成所有替换
result = s.replace(reverse_map)
print(result)

输出结果 #

0    odd
1    even
2    odd
3    even
4    odd
dtype: object

优势说明 #

• 扩展性强：后续要加新分组（比如"three_multiples": [3,6,9]），只需要在custom_groups里新增键值对，无需修改replace相关代码
• 代码更简洁：避免了链式调用多个replace，逻辑更清晰
• 性能更优：减少了Series的重复遍历操作

内容的提问来源于stack exchange，提问作者Howard