Python中添加元素后，如何修正有序集合的更新输出格式？

问题分析 #

当前代码存在两个核心问题导致输出不符合预期：

1. 使用split(",")处理单个输入值时，会返回包含该值的列表（比如输入4会得到['4']），直接插入主列表会产生嵌套结构。
2. 部分代码逻辑冗余，比如循环遍历data1添加元素的操作可以简化。

修改后的代码 #

# 处理data1和element的添加逻辑
data1 = input("data1: ").split(",")
element = input("element: ")
l1 = data1.copy()  # 直接复制data1元素，替代循环添加
l1.insert(-1, element)
print("sorted set after adding:", l1)

# 处理data2的更新逻辑
data2 = input("data2: ").strip()  # 直接获取输入并去除首尾空白，避免生成列表
l1.insert(0, data2)
print("sorted set after updating:", l1)

关键修改点 #

• 用l1 = data1.copy()替代循环遍历添加元素，简化代码逻辑。
• 处理data2时去掉split(",")，改用strip()处理输入，直接获取单个元素，避免生成嵌套列表。
• 插入data2时直接插入单个元素，解决输出中的嵌套问题。

可选：输出整数类型集合 #

如果需要输出[4, 3, 2]这种整数格式，可将输入转换为整数类型，代码调整如下：

data1 = [int(x) for x in input("data1: ").split(",")]
element = int(input("element: "))
l1 = data1.copy()
l1.insert(-1, element)
print("sorted set after adding:", l1)

data2 = int(input("data2: ").strip())
l1.insert(0, data2)
print("sorted set after updating:", l1)

内容的提问来源于stack exchange，提问作者Stanford J Gadaga