python两列数据的所有组合
常见问答
如何用Python生成两列数据的所有可能组合？
我有两列数据，想要生成这两列中每个元素之间的所有可能组合，应该用什么方法？
利用itertools.product生成两列数据的组合
可以使用Python标准库中的itertools.product函数，它能够生成输入数据的笛卡尔积，从而得到所有可能的元素组合。示例如下：
import itertools
list1 = [1, 2, 3]
list2 = ['a', 'b']
combinations = list(itertools.product(list1, list2))
print(combinations)
这样会输出[(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b'), (3, 'a'), (3, 'b')]，即所有两列数据的组合。
能否使用Pandas对两列数据求组合？
我使用Pandas处理数据，想得到两列中所有的组合项，怎样用Pandas做到？
Pandas中通过merge和cross join生成组合
Pandas可以通过merge实现笛卡尔积来生成两列数据的所有组合。具体方法为对同一个DataFrame做cross join，或者分别创建两个DataFrame后使用merge，如：
import pandas as pd
df1 = pd.DataFrame({'col1': [1, 2, 3]})
df2 = pd.DataFrame({'col2': ['a', 'b']})
# cross join
result = df1.assign(key=1).merge(df2.assign(key=1), on='key').drop('key', axis=1)
print(result)
上述代码可得到所有col1和col2的组合。
怎样避免生成重复或无效的组合？
生成两列数据所有组合时，如何确保不出现重复对或者无效组合？
通过数据清洗和条件筛选避免重复和无效组合
在生成组合前，应对原始数据去重，确保数据本身无重复元素。此外，生成组合后可以加入条件过滤，筛掉无效的数据对。例如，如果两列数据中存在相同元素但不希望两者匹配，可以排除两元素相等的组合。举例如下：
import itertools
list1 = [1, 2, 2]
list2 = [2, 3]
list1 = list(set(list1)) # 去重
combs = [ (a,b) for a, b in itertools.product(list1, list2) if a != b ]
print(combs)
这样既去掉了重复元素，也避免了某些无效组合。