用xlwt把文件名快速整理到Excel中

import os
import xlwt

file_dir = 'c:/'
os.listdir(file_dir)
new_workbook = xlwt.Workbook()
worksheet = new_workbook.add_sheet("new_test")
n = 0
for i in os.listdir(file_dir):
    worksheet.write(n, 0, i)
    n += 1
new_workbook.save(r"D:\mycode\files\报表2.xlsx")