#!/usr/bin/env python
# coding=utf-8
import glob
import time
import csv
# a new file
#open all the CSV file
#遍历文件夹下所有csv文件
csv_list = glob.glob('csvfile/*.csv')
print('共有%s个CSV文件'% len(csv_list))
print (csv_list)
# open every file in loop
for csv_file in csv_list:
csvreader = csv.reader(open(csv_file, "r"))
print (csvreader)
for index,rows in enumerate(csvreader):
#每个文件的第4行数据
if index == 3:
print (rows)
# write the data in a new file
#标头
fileHeader = ["nA", "nb", "nc"]
with open ('newfile.csv', 'a+') as wf:
new_writer = csv.writer(wf)
#new_writer.writerow(fileHeader)
new_writer.writerow(rows)
wf.close()
Glob 用法
glob模块实现对目录内容进行匹配,结合通配符星号(*)、问号(?)和中括号([ ])使用。星号可以匹配零个或多个符号,问号(?)可以匹配单个字符,中括号([ ])匹配一个指定范围的字符。
glob.glob() 函数接受通配模式作为输入,并返回所有匹配的文件名和路径名列表,与os.listdir()类似。