提取多个csv文件中的列组成新的csv文件的python代码
以下是一个示例代码,可以从多个CSV文件中提取所需列并将其组成新的CSV文件:
import pandas as pd
import glob
# 要提取的列名称列表
columns_to_extract = ['列名1', '列名2', '列名3']
# 要读取的CSV文件路径
path = '路径/到/CSV文件'
# 获取文件夹中所有CSV文件的路径
all_files = glob.glob(path + "/*.csv")
# 创建一个空的DataFrame来存储提取的数据
data = pd.DataFrame()
# 循环读取所有CSV文件并提取所需列
for filename in all_files:
df = pd.read_csv(filename, usecols=columns_to_extract)
data = pd.concat([data, df], axis=0, ignore_index=True)
# 将提取的数据保存为新的CSV文件
data.to_csv('新的CSV文件路径', index=False)
请注意,在此示例中,我们使用pandas库来读取和操作CSV文件。 glob库用于获取指定文件夹中所有CSV文件的路径。usecols参数用于指定要提取的列。pd.concat()函数用于将所有数据合并为一个DataFrame。最后,我们将新的DataFrame保存为新的CSV文件,使用index=False参数来避免将索引列保存到CSV文件中
原文地址: https://www.cveoy.top/t/topic/hbDD 著作权归作者所有。请勿转载和采集!