提取一个文件中的所有csv文件的指定列与行的数据,存储在dataframe格式中,最后把获取数据统一写入该文件路径下的result.csv中
import csv
import os
import pandas as pd
#处理函数
def file_name(file_dir):
L=[]
#取文件名,并存在L中
name = os.path.splitext(file_dir)[0]#获取文件夹名
source_dir=os.getcwd()#读取当前文件夹路径
for root, dirs,files in os.walk(file_dir): #获取文件夹下所有csv文件名
for file in files:
if os.path.splitext(file)[1] == '.csv':
L.append(os.path.join(root, file))
#获取每个文件的Sound的0-60行,存到datadf中
datadf = pd.DataFrame({'Sound':[]})
for i in range(len(L)):
print(L[i])
d = pd.read_csv(L[i],usecols = ['Sound'],nrows = 60)#可自己命名
datadf = pd.concat([datadf,d])
#在csv目录下生成新的csv文件
datadf.to_csv(source_dir + "\\" + name + "\\result" + ".csv",index=0,sep=',')
if __name__== "__main__":
file_name("文件路径")