CSV（逗号分隔值）文件广泛用于以简单格式存储和交换数据。在许多数据处理任务中，需要基于特定列合并两个或多个CSV文件。幸运的是，这可以使用 Python 中的 Pandas 库轻松实现。
在本文中，我们将学习如何使用 Python 中的 Pandas 按特定列合并两个 CSV 文件。
什么是 Pandas 库？
Pandas 是一个用于 Python 信息控制和检查的开源库。它提供了用于处理结构化数据（例如表格、时间序列和多维数据）以及高性能数据结构的工具。 Pandas 广泛应用于金融、数据科学、机器学习和其他需要数据操作的领域。
在 Python 中按特定列合并两个 CSV 文件的步骤
以下是使用 Pandas 库在 Python 中按特定列合并两个 CSV 文件的完整步骤 –
立即学习“Python免费学习笔记（深入）”；
第 1 步：导入 Pandas 库
合并两个 CSV 文件的第一步是导入 pandas 库。 Pandas 是一个强大的 Python 数据分析库。它提供了用于有效存储和操作大型数据集的数据结构。要使用 Pandas，我们首先需要将其导入到 Python 程序中。我们可以使用以下命令来做到这一点 –
import pandas as pd
登录后复制
第 2 步：读取 CSV 文件
我们的下一步是读取我们想要合并的两个 CSV 文件。我们可以使用 Pandas 的 read_csv() 函数将 CSV 文件读取到 Pandas DataFrame 中。我们需要提供 CSV 文件的文件路径或 URL 作为 read_csv() 函数的参数。例如 –
df1 = pd.read_csv('file1.csv')df2 = pd.read_csv('file2.csv')
登录后复制
第 3 步：合并 CSV 文件
将 CSV 文件读入 Pandas DataFrames 后，现在是时候使用 merge() 函数根据特定列合并它们了。 merge() 函数将两个 DataFrame 作为输入，并基于公共列将它们合并。这是基本语法 –
merged_df = pd.merge(df1, df2, on='column_name')
登录后复制
在此示例中，merged_df 是新的 DataFrame，其中包含 df1 和 df2 的合并数据。 on 参数指定合并两个 DataFrame 的公共列的名称。
例如，假设我们有两个 CSV 文件 sales.csv 和customers.csv。这两个文件都包含名为 CustomerID 的列。我们希望根据 CustomerID 列合并两个文件。我们可以这样做 –
sales = pd.read_csv('sales.csv')customers = pd.read_csv('customers.csv')merged_df = pd.merge(sales, customers, on='CustomerID')
登录后复制
第 4 步：处理缺失值（如果有）
根据特定列合并两个 CSV 文件时，合并后的 DataFrame 中可能会缺少值。如果其中一个 CSV 文件的相应列中没有匹配的值，则可能会出现这些缺失值。
为了处理缺失值，我们可以使用 Pandas 的 fillna() 函数将其替换为默认值或计算值。例如，我们可以使用以下代码将缺失值替换为字符串“Unknown” –
merged_df.fillna('Unknown', inplace=True)
登录后复制
在此示例中，我们使用 fillna() 函数将合并的 DataFrame 中的缺失值替换为字符串“Unknown”。我们还指定了 inplace 参数为 True，这意味着将修改原始 DataFrame，而不是创建新的 DataFrame。
示例 1：使用 Merge() 函数
在此示例中，我们将使用 Pandas 的 merge() 函数根据特定列合并两个 CSV 文件。
CSV Files
登录后复制登录后复制
假设我们有两个 CSV 文件：employees.csv 和 Departments.csv。 Employees.csv 文件包含以下数据 –
员工ID 姓名 部门ID 薪资1约翰1500002莎拉2600003大卫1550004亚历克斯3650005艾米丽255000
departments.csv 文件包含以下数据 –
部门ID 部门名称1IT2销售3营销4人力资源
示例
import pandas as pd# Load CSV Filesemployees = pd.read_csv('employees.csv')departments = pd.read_csv('departments.csv')# Merge dataframes based on DepartmentID columnmerged_df = pd.merge(employees, departments, on='DepartmentID')# Print merged dataframeprint(merged_df.head())# Save merged dataframe to a new CSV filemerged_df.to_csv('merged_employees_departments.csv', index=False)
登录后复制
输出
EmployeeID Name DepartmentID Salary DepartmentName0 1 John 1 50000 IT1 3 David 1 55000 IT2 2 Sarah 2 60000 Sales3 5 Emily 2 55000 Sales4 4 Alex 3 65000 Marketing
登录后复制
示例 2：使用 Join() 函数
在本例中，我们将使用 Pandas 的 join() 方法根据特定列合并两个 CSV 文件。
CSV Files
登录后复制登录后复制
假设我们有两个 CSV 文件：orders.csv 和customers.csv。 order.csv 文件包含以下数据 –
订单ID 客户ID 订单日期 总金额112022-05-01100.0232022-05-02150.0322022-05-03200.0412022-05-0475.0542022-05-05120.0
customers.csv 文件包含以下数据 –
客户ID 客户名称 电子邮件1约翰john@example.com2莎拉sarah@example.com3大卫david@example.com4艾米丽emily@example.com
示例
import pandas as pd# Load CSV filesorders = pd.read_csv('orders.csv')customers = pd.read_csv('customers.csv')# Join dataframes based on CustomerID columnjoined_df = orders.set_index('CustomerID').join(customers.set_index('CustomerID'))# Print joined dataframeprint(joined_df.head())# Save joined dataframe to a new CSV filejoined_df.to_csv('joined_orders_customers.csv')
登录后复制
输出
OrderID OrderDate TotalAmount CustomerName EmailCustomerID 1 1 2022-05-01 100.0 John john@example.com1 4 2022-05-04 75.0 John john@example.com2 3 2022-05-03 200.0 Sarah sarah@example.com3 2 2022-05-02 150.0 David david@example.com4 5 2022-05-05 120.0 Emily emily@example.com
登录后复制
使用 Pandas 的 merge() 函数，我们根据本例中的“id”列合并了两个 CSV 文件。作为合并两个 CSV 文件的结果的 DataFrame 除了“name_x”、“email_x”、“name_y”和“email_y”列之外还包括“id”列。
请注意，“name_y”和“email_y”段的组合 DataFrame 中缺少值，这些值与第二个 CSV 记录中没有匹配质量的行相关。如上一步所示，Pandas fillna() 和 dropna() 函数可用于处理这些缺失值。
结论
基于特定列合并两个 CSV 文件是一项常见的数据处理任务，可以使用 Python 中的 Pandas 库轻松实现。在本文中，我们学习了如何使用 Pandas 的 merge() 函数合并两个 CSV 文件。我们还讨论了如何处理缺失值以及如何将合并的 DataFrame 保存到新的 CSV 文件。
以上就是如何使用Python中的Pandas按特定列合并两个CSV文件？的详细内容，更多请关注【创想鸟】其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至253000106@qq.com举报，一经查实，本站将立刻删除。
发布者：PHP中文网，转转请注明出处：https://www.chuangxiangniao.com/p/2224817.html
