读取文件夹的所有CSV文件,都取第三行写入新文件。


#!/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()类似。

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐