根据第一列中的数据合并两个CSV文件 [英] Merge two CSV files based on a data from the first column

问题描述

我有两个要合并的csv文件，如下所示-或多或少地使用第一列ID_作为唯一标识符，并将AMT列附加到最终文件中的新行.

I have two csv files like below that I'd like to merge - more or less using the first column ID_ as the unique identifier, and append the AMT column to a new row in the final file.

CSV1

ID_ CUSTOMER_ID_    EMAIL_ADDRESS_
1090    1   [e-mail]
1106    2   [e-mail]
1145    3   [e-mail]
1206    4   [e-mail]
1247    5   [e-mail]
1254    6   [e-mail]
1260    7   [e-mail]
1361    8   [e-mail]
1376    9   [e-mail]

CSV2

ID_ AMT
1090    5
1106    5
1145    5
1206    5
1247    5
1254    65
1260    5
1361    10
1376    5

这是我在最终文件中寻找的内容:

Here's what I'm looking for in a final file:

ID_ CUSTOMER_ID_    EMAIL_ADDRESS_  AMT
1090    1   [e-mail]    5
1106    2   [e-mail]    5
1145    3   [e-mail]    5
1206    4   [e-mail]    5
1247    5   [e-mail]    5
1254    6   [e-mail]    65
1260    7   [e-mail]    5
1361    8   [e-mail]    10
1376    9   [e-mail]    5

我尝试过尽可能地在下面对此进行修改，但无法获得所需的内容.真的卡在了这里-不知道我还能做什么.非常感谢任何帮助！

I've tried modifying a this below as much as possible, but not able to get what I'm looking for. Really stuck on this - not sure what else I can do. Really appreciate any and all help!

join -t, File1.csv File2.csv

此示例中显示的数据包含选项卡，但我的实际文件是上述的CSV文件，并且将逗号作为分隔符.

Data shows in this example contains tabs, but my actual files are CSVs as mentioned and will contain commas as a separator.

推荐答案

使用Pandas库可以轻松完成此操作.这是我执行此操作的代码:

This can be easily done using Pandas library. Here is my code to do this:

'''
This program reads two csv files and merges them based on a common key column.
'''
# import the pandas library
# you can install using the following command: pip install pandas

import pandas as pd

# Read the files into two dataframes.
df1 = pd.read_csv('CSV1.csv')
df2 = pd.read_csv('CSV2.csv')

# Merge the two dataframes, using _ID column as key
df3 = pd.merge(df1, df2, on = 'ID_')
df3.set_index('ID_', inplace = True)

# Write it to a new CSV file
df3.to_csv('CSV3.csv')

您可以在此处找到有关熊猫的简短教程: https://pandas.pydata.org/pandas-docs/stable/getting_started/10min.html

You can find a short tutorial on pandas here: https://pandas.pydata.org/pandas-docs/stable/getting_started/10min.html

这篇关于根据第一列中的数据合并两个CSV文件的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！