程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
您现在的位置: 程式師世界 >> 編程語言 >  >> 更多編程語言 >> Python

python批量提取指定列並寫入新文件

編輯:Python
# -*- 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)


  1. 上一篇文章:
  2. 下一篇文章:
Copyright © 程式師世界 All Rights Reserved