将dataframe1, 和dataframe2 按照列code合并,dataframe只要保留列A和列B
时间: 2024-10-30 20:02:23 浏览: 25
python 给DataFrame增加index行名和columns列名的实现方法
在Pandas中,如果你想要将两个DataFrame `dataframe1` 和 `dataframe2` 按照列 'code' 合并,并且只保留这两个DataFrame中的列 'A' 和 'B',你可以使用 `merge` 函数,并设置 `'on'` 参数为 'code'。下面是一个例子:
```python
import pandas as pd
# 假设 dataframes 的结构类似这样
dataframe1 = {
'code': ['Code1', 'Code2', 'Code3'],
'A': [1, 2, 3],
'C': [4, 5, 6]
}
dataframe2 = {
'code': ['Code2', 'Code3', 'Code4'],
'B': [7, 8, 9],
'D': [10, 11, 12]
}
# 创建 DataFrames
df1 = pd.DataFrame(dataframe1)
df2 = pd.DataFrame(dataframe2)
# 合并 DataFrame 并选择需要的列
merged_df = pd.merge(df1[['code', 'A']], df2[['code', 'B']], on='code')
print(merged_df)
```
这将返回一个新的 DataFrame,其中包含了 'code', 'A', 和 'B' 列的内容,同时基于 'code' 进行了合并。
阅读全文