'''写入excel文件'''
import xlsxwriter

# todo 创建excel文件
xl = xlsxwriter.Workbook(r'D:\testfile\test.xlsx')

# todo 添加sheet
sheet = xl.add_worksheet('sheet1')

# todo 往单元格cell添加数据,索引写入
sheet.write_string(0,0,'username')

# todo 位置写入
sheet.write_string('B1','password')

# todo 设置单元格宽度大小
sheet.set_column('A:B',30)

# todo 关闭文件
xl.close()
import os
import openpyxl


def get_file_name(file_dir):
'''
获取指定目录下所有文件名称
:param file_dir:指定目录
:return:返回文件名列表
'''
for root, dirs, files in os.walk(file_dir):
# return root#当前目录路径
# return dirs#当前路径下所有子目录
return files # 当前路径下所有非目录子文件


def class_file_name(file_dir, file_type):
'''
获取指定文件夹下的指定文件类型名称
:param file_dir:指定目录
:param file_type:
:return:返回文件名列表
'''
ls = []
f = []
for root, dirs, files in os.walk(file_dir):
for file in files:
if os.path.splitext(file)[1] == file_type:
ls.append(os.path.join(root, file))
f.append(file)
return f


def output2excel(file_dir):
'''
把文件夹下的文件名称输出到文件目录
:param file_dir: 文件目录
:return:
'''
# 获取文件目录下所有文件名,存入data列表
data = get_file_name(file_dir)

# 把data输出到该目录下,并以目录名保存为excel格式
wb = openpyxl.Workbook()
sheet = wb.active
# 设置表名为文件目录名
sheet.title = file_dir
for i in range(1, len(data) + 1):
sheet['A{}'.format(i)] = data[i - 1]

if file_dir == '':
file_dir = '当前目录'
wb.save('{0}/{0}.xlsx'.format(file_dir))


file_dir = '本科'
output2excel(file_dir)

 

心有猛虎,细嗅蔷薇