pandas 学习

读取csv文件并选择行合并

知识点:

os 读取文件夹和文件名

import pandas as pd
import os  

Folder_Path = r'C:\Users。。。'  # 要拼接的文件夹及其完整路径,注意不要包含中文
SaveFile_Path = r'C:\Users。。。'  # 拼接后要保存的文件路径
SaveFile_Name = r'all.csv'  # 合并后要保存的文件名

# 修改当前工作目录
os.chdir(Folder_Path)
# 将该文件夹下的所有文件名存入一个列表
file_list = os.listdir()

# 读取第一个CSV文件并包含表头
df = pd.read_csv(Folder_Path + '\\' + file_list[0])  # 编码默认UTF-8,若乱码自行更改

# 将读取的第一个CSV文件写入合并后的文件保存
df.loc[14:19].to_csv(SaveFile_Path + '\\' + SaveFile_Name, encoding="utf_8_sig", index=False)

# 循环遍历列表中各个CSV文件名,并追加到合并后的文件
for i in range(1, len(file_list)):
    df = pd.read_csv(Folder_Path + '\\' + file_list[i])
    df.loc[14:19].to_csv(SaveFile_Path + '\\' + SaveFile_Name, encoding="utf_8_sig", index=False, header=False, mode='a+')

 

posted @ 2022-05-27 20:26  yayayua  阅读(35)  评论(0)    收藏  举报