📜  python - 仅合并和包含特定列 - Python (1)

📅  最后修改于: 2023-12-03 14:45:53.413000             🧑  作者: Mango

Python - 仅合并和包含特定列

在Python中,有时候你可能需要处理一个大型文本文件或CSV文件,而只需要合并和包含特定列。这通常通过使用Pandas和csv模块来完成。下面是实现这个目标的代码片段。

import csv
import pandas as pd

# 读取CSV文件
df = pd.read_csv('filename.csv')

# 保留所需列的数据
df = df[['Column1', 'Column2', 'Column3']]

# 写入CSV文件
df.to_csv('newfilename.csv', index=False)

首先,我们使用Pandas的read_csv函数读取CSV文件。然后,我们使用DataFrame中的列名来仅保留需要的列。最后,我们使用to_csv函数将数据写入新的CSV文件中。

请注意,index=False参数用于排除DataFrame的索引列。

这是一个简单而有效的方法来合并和包含特定列的大型CSV文件。