如何将Pandas DataFrame转换为含唯一整数对与权重字典的元组列表

嘿，这问题不难搞定，我给你个直接高效的方案！

我们可以借助Python的itertools.combinations来快速生成ID列的所有不重复两两组合，再通过列表推导式把每个组合和指定的weight字典打包成你要的元组格式。

步骤1：先构造示例DataFrame（如果还没创建的话） #

import pandas as pd
df = pd.DataFrame({'ID': [2,3,4,5,6,7], 'weight': [1]*6})

步骤2：生成目标元组列表 #

itertools.combinations会自动生成所有长度为2的不重复ID组合（它不会生成反向重复的组合，比如不会同时出现(2,3)和(3,2)，完全匹配你给出的示例顺序），再用列表推导式把每个组合和固定的weight字典组装成目标元组：

from itertools import combinations

# 生成符合要求的元组列表
result = [(a, b, {'weight': 1}) for a, b in combinations(df['ID'], 2)]

验证输出效果 #

打印前几个元素看看是否符合预期：

print(result[:5])
# 输出结果：
# [(2, 3, {'weight': 1}), (2, 4, {'weight': 1}), (2, 5, {'weight': 1}), (2, 6, {'weight': 1}), (2, 7, {'weight': 1})]

这个方法简洁又高效，不管你的ID数量有多少，itertools的组合生成逻辑都是经过优化的，能轻松应对需求。

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