Python合并角度与距离列表时丢失数据点的问题排查与解决

嘿，你确实碰到数据丢失的问题了！核心原因是字典的键是唯一的——当你用dict(zip(angle, dist))把角度和距离转成字典时，如果原angle列表里存在重复的角度值，后面的键值对会直接覆盖掉前面的。比如如果原列表里有两个相同的角度30，对应的距离分别是10和15，转成字典后只会保留最后一个30:15，自然就丢了一个数据点，这就是两个列表长度差很多的原因。

最稳妥的方式是跳过字典，直接把角度和距离配对成元组列表，然后按角度排序，再拆分回两个列表，这样能完整保留所有原始数据点：

# 将角度和距离打包成元组列表，每个元组对应一个原始数据点
paired_data = list(zip(angle, dist))
# 按元组的第一个元素（角度值）排序
paired_data.sort(key=lambda item: item[0])
# 拆分排序后的元组列表，得到有序的角度和距离列表
sorted_angle, sorted_dist = zip(*paired_data)
# 如果需要将结果转为列表类型（zip返回的是元组），可以再加一步转换
sorted_angle = list(sorted_angle)
sorted_dist = list(sorted_dist)

这个方法的好处是，哪怕有重复的角度值，每个原始数据点的配对关系都会被保留，排序后数据点的数量和原列表完全一致。

如果你确实需要用字典来整理同角度的多个距离数据，可以用collections.defaultdict来存储：

from collections import defaultdict

# 创建一个默认值为列表的字典，用来存每个角度对应的所有距离
angle_dist_dict = defaultdict(list)
for a, d in zip(angle, dist):
    angle_dist_dict[a].append(d)

# 按角度值排序键
sorted_angles = sorted(angle_dist_dict.keys())

# 展开成和原列表长度一致的有序列表
sorted_angle = []
sorted_dist = []
for angle_val in sorted_angles:
    # 把每个角度对应的所有距离依次加入列表
    for dist_val in angle_dist_dict[angle_val]:
        sorted_angle.append(angle_val)
        sorted_dist.append(dist_val)

这个方法适合需要按角度分组查看数据的场景，同样能保留所有原始数据点。

内容的提问来源于stack exchange，提问作者João Roquette Saldanha