生成txt

import os
def generate(dir,label):
    files = os.listdir(dir)
    files.sort()
    print ('****************')
    print ('input :',dir)
    print ('start...')
    listText = open(dir+'/'+'list.txt','w')
    for file in files:
        fileType = os.path.split(file)
        if fileType[1] == '.txt':
            continue
        name = file + ' ' + str(int(label)) +'\n'
        listText.write(name)
    listText.close()
    print ('down!')
    print ('****************')

if __name__ == '__main__':
    generate('/Users/cs/Desktop/paper/第二学期/deepclusering/data/1',0)