查看文件夹内的文件数并写入txt

import os

target_path='/content/drive/MyDrive/srntt-pytorch-master/data/CUFED/ref'

target_path2='/content/drive/MyDrive/srntt-pytorch-master/data/CUFED/input'

all_content=os.listdir(target_path)

all_content2=os.listdir(target_path2)

print('All ref numbers is',len(all_content))

print('All input numbers is',len(all_content2))

# count_num=1

# file = open('/content/drive/MyDrive/srntt-pytorch-master/data/CUFED/b.txt','w')

# for content in all_content2:

#     file.write(str(count_num) + '---------'+content + '\n')

#     count_num+=1