python提取文件名列表学号部分

python提取文件名列表学号部分，并生成新的CSV文件。

说明

数据源为文件夹下的很多文件的文件名，因为文件名的特殊性，python提取标识文件所有者的学号信息，从文件命名中提取，并生成学号的CSV文件。

code

import os
import re
import csv

# 输出文件，输出到项目文件夹
file_out = 'xxx.csv'

# 输出文件的字段名
header = ['学号']

# 输出文件写header
f_out = open(file_out, 'w', newline='')
writer = csv.DictWriter(f_out, header)
writer.writeheader()

# 文件夹绝对路径
filePath = 'xx\\xx\\xx\\'

# 读取所有的文件名到列表
fileList = os.listdir(filePath)

# 文件总数
total = len(fileList)
# print(total)

# 文件数计数
count = 0

# 遍历文件名
for fileName in fileList:
    # 文件数计数器
    count += 1
    # print(count)
    # 正则表达式，提取学号部分
    # 因为学号是连续的数字，方便识别提取
    # 正则表达式中d表示数字，13表示数字部分的长度
    studentNum = re.findall(r'\d{13}', fileName)
    # 异常信息不处理，比如学号位数输错
    if len(studentNum) == 0:
        writer.writerow({'学号': "异常：" + fileName + '\t'})
        continue
    # 打印学号信息，校对正确性
    # print(studentNum[0])
    # 写数据到输出文件，'\t'避免科学计数法显示
    writer.writerow({'学号': studentNum[0] + '\t'})
    # 处理文件数与文件总数是否一致
    if count == total:
        print("已正确处理完所有文件")

f_out.close()