Python数据处理 | 批量提取文件夹下的csv文件,每个csv文件根据列索引提取特定几列,并将提取后的数据保存到新建的一个文件夹
文章目录
- 一、问题来源
- 二、解决方法
一、问题来源
价值
元的问答,那天在准备去吃饭前刚好看到,几分钟搞定,午饭加个鸡腿~~
二、解决方法
实现代码如下:
import os
import pandas as pd
path1 = "你放所有csv的文件夹路径" # 你放所有csv的文件夹路径
path2 = "./data" # 新建一个文件夹 文件夹名data 当前目录下 你也可以指定
if not os.path.exists(path2):
os.mkdir(path2)
for filename in os.listdir(path):
# 是csv文件
if filename.endswith(".csv"):
file_path1 = path1 + "/" + filename
# 读取csv可能会编码错误 还可加参数 engine="python" 或者指定编码 encoding="utf-8"就可以解决
df1 = pd.read_csv(file_path1)
# 索引指定列的数据
df2 = df1[['时间', '风机', '平均齿轮箱主滤芯1_1压力',
'平均齿轮箱主滤芯1_2压力', '平均齿轮箱主滤芯2_1压力',