python - 两列可能组合的并集

Question

我的 DataFrame 看起来像这样:

A    B
100  1
100  2
200  2
200  3

我需要找到 A 和 B 值的所有可能组合，并使用这些组合创建新的数据框，第三列指示原始 df 中存在的每个组合:

A   B  C
100 1  True
100 2  True
100 3  False
200 1  False
200 2  True
200 3  True

我现在是怎么做的:

import pandas as pd

df = pd.DataFrame({'A' : [100,100,200,200], 'B' : [1,2,2,3]})
df['D'] = 42
df2 = df[['A','D']].merge(df[['B','D']], on = 'D')
  [['A','B']].drop_duplicates()
i1 = df.set_index(['A','B']).index
i2 = df2.set_index(['A','B']).index
df2['C'] = i2.isin(i1)
print(df2)

它有效，但看起来很丑。有更清洁的方法吗？

Answer 1

您可以使用:

• 创建填充True的新列
• set_index来自所有组合的列
• 创建MultiIndex.from_product来自 df1 索引
• reindex原始 df 如果不存在则添加 Falses
• reset_index对于 MultiIndex
中的列

---

df['C'] = True
df1 = df.set_index(['A','B'])
mux = pd.MultiIndex.from_product(df1.index.levels, names=df1.index.names)
df = df1.reindex(mux, fill_value=False).reset_index()
print (df)
     A  B      C
0  100  1   True
1  100  2   True
2  100  3  False
3  200  1  False
4  200  2   True
5  200  3   True