插件窝 干货文章 如何高效合并两个DataFrame:基于索引的巧妙连接?

如何高效合并两个DataFrame:基于索引的巧妙连接?

DataFrame 方法 合并 index 760    来源:    2025-03-13

在Pandas中,合并两个DataFrame可以通过多种方式实现,具体取决于你想要如何连接它们。如果你希望基于索引进行合并,可以使用以下几种方法:

1. 使用 join 方法

join 方法是基于索引合并DataFrame的常用方法。默认情况下,join 会基于索引进行左连接。

import pandas as pd

# 创建两个示例DataFrame
df1 = pd.DataFrame({'A': [1, 2, 3]}, index=['a', 'b', 'c'])
df2 = pd.DataFrame({'B': [4, 5, 6]}, index=['a', 'b', 'd'])

# 使用join方法合并
result = df1.join(df2, how='inner')  # 可以根据需要选择how参数:'left', 'right', 'outer', 'inner'
print(result)

2. 使用 merge 方法

merge 方法也可以基于索引进行合并,但需要显式指定 left_indexright_index 参数。

import pandas as pd

# 创建两个示例DataFrame
df1 = pd.DataFrame({'A': [1, 2, 3]}, index=['a', 'b', 'c'])
df2 = pd.DataFrame({'B': [4, 5, 6]}, index=['a', 'b', 'd'])

# 使用merge方法合并
result = pd.merge(df1, df2, left_index=True, right_index=True, how='inner')  # 可以根据需要选择how参数
print(result)

3. 使用 concat 方法

concat 方法可以沿着指定的轴(通常是行或列)连接多个DataFrame。如果你想基于索引进行连接,可以设置 axis=1

import pandas as pd

# 创建两个示例DataFrame
df1 = pd.DataFrame({'A': [1, 2, 3]}, index=['a', 'b', 'c'])
df2 = pd.DataFrame({'B': [4, 5, 6]}, index=['a', 'b', 'd'])

# 使用concat方法合并
result = pd.concat([df1, df2], axis=1, join='inner')  # 可以根据需要选择join参数:'inner', 'outer'
print(result)

4. 使用 combine_first 方法

combine_first 方法可以用于填充缺失值,它会用第二个DataFrame中的值填充第一个DataFrame中的缺失值。

import pandas as pd

# 创建两个示例DataFrame
df1 = pd.DataFrame({'A': [1, 2, 3]}, index=['a', 'b', 'c'])
df2 = pd.DataFrame({'A': [4, 5, 6]}, index=['a', 'b', 'd'])

# 使用combine_first方法合并
result = df1.combine_first(df2)
print(result)

总结

  • join:适用于基于索引的简单合并。
  • merge:适用于更复杂的合并操作,可以指定多个键。
  • concat:适用于沿着轴连接多个DataFrame。
  • combine_first:适用于填充缺失值。

根据你的具体需求选择合适的方法。如果你只需要基于索引进行简单的合并,join 方法通常是最直接和高效的选择。