python批量提取指定列并写入新文件

weixin_51443598

已于 2022-06-21 16:36:48 修改

阅读量457

点赞数

文章标签： python pandas 机器学习

于 2022-06-21 16:29:45 首次发布

本文链接：https://blog.csdn.net/weixin_51443598/article/details/125393784

版权

# -*- coding:utf-8 -*-
"""
作者：sunli
日期：2022年06月21日14:49
"""
# -*- coding:utf-8 -*-

import os
import numpy as np
import pandas as pd

path0 = r"D:\shujuji\SisFall_dataset"
path1 = r"D:\shujuji\SisFall_dataset" + '\\'

# 列出当前目录下所有的文件
files = os.listdir(path0)
print('files', files)
for filename in files:
    path = path1 + filename
    file = os.listdir(path)
    for file_ in file:
        portion = os.path.splitext(file_)
        # 如果后缀是.csv
        if portion[1] == ".csv":
            filenamedir = path + '\\'+ file_
            data = pd.read_csv(filenamedir, index_col=False, header=None)
            data_size = len(data)
            col_data = np.arange(0, data_size, 1)
            extract_data = data.iloc[col_data, [0, 1, 2, 3, 4, 5]]
            extract_data.columns = ["acc_x", "acc_y", "acc_z", "gyro_x", "gyro_y", "gyro_z"]
            extract_data.to_csv(filenamedir, index=None)