DataFrame中如何进行SQL联结操作 - 问答

在Python的pandas库中，可以使用merge()函数实现类似SQL中的联结（join）操作

首先，我们需要导入pandas库并创建两个DataFrame。这里我们创建两个简单的DataFrame，分别为df1和df2：

import pandas as pd

data1 = {'key': ['A', 'B', 'C', 'D'],
         'value': [1, 2, 3, 4]}

data2 = {'key': ['B', 'D', 'E', 'F'],
         'value': [5, 6, 7, 8]}

df1 = pd.DataFrame(data1)
df2 = pd.DataFrame(data2)

接下来，我们可以使用merge()函数将这两个DataFrame进行联结。默认情况下，merge()函数会根据相同的列名进行内部联结（inner join）。例如，我们可以使用以下代码将df1和df2根据’key’列进行联结：

result = pd.merge(df1, df2, on='key')
print(result)

输出结果如下：

  key  value_x  value_y
0   B        2        5
1   D        4        6

如果想要进行其他类型的联结，例如左联结（left join）、右联结（right join）或全外联结（full outer join），可以通过设置how参数实现。例如，以下代码演示了如何进行左联结：

result = pd.merge(df1, df2, on='key', how='left')
print(result)

输出结果如下：

  key  value_x  value_y
0   A        1      NaN
1   B        2      5.0
2   C        3      NaN
3   D        4      6.0

注意，在这个例子中，由于’key’列中的’A’和’C’在df2中不存在，因此对应的value_y值为NaN。

0 赞

0 踩