上班这点事

如何抓取文件名到Excel表格中

2020-04-11  本文已影响0人  Hunter_Coming
  1. 将所需要的文件集中放到一个文件夹中

  2. 找到该文件夹的路径

  3. 运行下面代码

import os
import csv


class GetFileNames(object):
    def __init__(self):
        self.title_name = []
        self.path = "输入文件夹路径"

    def file_name(self):
        for root, dirs, files in os.walk(self.path):
            for file in files:
                self.title_name.append(os.path.splitext(file)[0])
        #print(self.title_name)

    def name_to_csv(self):
        a = []
        b = []
        csv_fp = open("你创建的csv文件的文件名", "w")
        sheet_title = ["Title"]
        writer = csv.writer(csv_fp)
        writer.writerow(sheet_title)
        for c in self.title_name:
            a.append(c)
            a=[]
            b.append(a)
        #print(b)
        writer.writerows(b)
        csv_fp.close()

    def run(self):
        self.file_name()
        self.name_to_csv()


GetFileNames().run()

然后就会生成一个n*1的列表,1列n(文件数)行。

上一篇 下一篇

猜你喜欢

热点阅读