Python如何合并两个csv文件

Question

我正在尝试将两个 csv 文件（a 和 b）合并为一个（c）。 csv 文件如下所示：

--- CSV A ---

Field_1, Key
a0     , k0
a1     , k1
a2     , k2
a3     , k0
a4     , k2
a5     , k0

--- CSV B ---

Key, Field_2, Field_3
k0 , b0     , c0
k1 , b1     , c1
k2 , b2     , c2
k3 , b3     , c3

--- 预期的 CSV C（合并的 csv）---

Field_1, Key, Field_2, Field_3
a0     , k0 , b0     , c0
a1     , k1 , b1     , c1
a2     , k2 , b2     , c2
a3     , k0 , b0     , c0
a4     , k2 , b2     , c2
a5     , k0 , b0     , c0

所以基本上应该连接 csv b 中与 csv a 的键匹配的字段以获得 csv c。但相反，我让下一个合并的字段为空

--- 实际 CSV C ---

Field_1, Key, Field_2, Field_3
a0     , k0 ,        , 
a1     , k1 ,        , 
a2     , k2 ,        , 
a3     , k0 ,        , 
a4     , k2 ,        , 
a5     , k0 ,        , 

这是我试图用来合并这些字段的代码。但正如我所说，我无法从合并的 csv b 中获取数据，我只能获取标题。

a = pd.read_csv("a.csv")
b = pd.read_csv("b.csv").rename(columns={'Key': ' Key'})
result = a.merge(b, on=" Key", how="left")
result.to_csv("c.csv", index=False)

那么我该怎么做才能正确合并来自 csv b 的数据？谢谢。

Answer 1

问题是您的关键字段中有空格（_）。在 a.csv 中有“_key”（例如“k0”），在 b.csv 中有“key”（例如“k0_”），因此密钥不匹配。如果您删除 csv 文件中的空格，则此代码有效：

import pandas as pd

a = pd.read_csv("a.csv",sep=",")
b = pd.read_csv("b.csv",sep=",")

pd.merge(a,b,on="Key",how="left")

您可以在导入 a.csv 期间使用 skipinitialspace=True，因为空格在前面，如下所示：

a = pd.read_csv("a.csv",sep=",",skipinitialspace=True)

或者你定义一个去除任何空白的函数并在导入时应用它：

def trim(dataset):

    trim = lambda x: x.strip() if type(x) is str else x #Stripping whitespaces in values

    dataset = dataset.rename(columns=lambda x: x.strip()) #Stripping whitespaces in colnames

    return dataset.applymap(trim)

a = trim(pd.read_csv("a.csv",sep=","))
b = trim(pd.read_csv("b.csv",sep=","))

Answer 2

只需将how 属性更改为outer。

result = a.merge(b, on=" Key", how = "outer")

left 仅使用左侧框架中的keys，类似于 SQL 左外连接；保留键顺序

Answer 3

由于key不正确，合并后的数据为空。根据您在 b.csv 中的数据键与 a.csv 中的键不同，因为它包含一个额外的空间。

您的代码将适用于这些数据

a.csv

Field_1,Key a0, k0 a1, k1 a2, k2 a3, k0 a4, k2 a5, k0

b.csv

Key, Field_2, Field_3 k0, b0, c0 k1, b1, c1 k2, b2, c2 k3, b3, c3