将一个文件夹下所有的文件夹的名称提取并写入Excel中

在工作的时候，有时候会需要获取文件夹下的所有文件名，文件名下的具体文件不用动。

import os
import xlwt

#"将一个文件夹下所有的文件夹的名称提取并写入Excel中"
workbook = xlwt.Workbook()
worksheet = workbook.add_sheet('pages')
rootdir = r"E:\新建文件夹"
i = 0
for dirnames in os.listdir(rootdir):
    worksheet.write(i,0,dirnames)
    i+=1
workbook.save(r'E:\新建文件夹\0组2dirnames.xls')
print(i)

在该文件夹下创建一个Excel，里面写入提取的文件夹的名，再打印一下一共提取了多少个。