得到两个dataframe主键的交集
时间: 2024-09-26 21:08:37 浏览: 25
Pandas DataFrame Notes
在Python中,如果你有两个`pandas` DataFrame并且你想找到它们主键(通常指`index`或`key`列)的交集,你可以使用`merge`函数并设置`how='inner'`参数。这将返回两个DataFrame共享的行。这里有一个例子:
```python
import pandas as pd
# 假设 df1 和 df2 是你的两个DataFrame
df1 = pd.DataFrame(..., index=['a', 'b', 'c'])
df2 = pd.DataFrame(..., index=['b', 'd', 'e'])
# 使用 inner join 获取交集
intersection_df = pd.merge(df1, df2, left_index=True, right_index=True, how='inner')
print(intersection_df)
```
在这个例子中,`left_index=True` 和 `right_index=True` 表示我们使用的是DataFrame的索引作为键进行合并,如果它们的索引不匹配,那么结果就会是空的。
阅读全文