python对文件批量打标签

功能：给数据打标签

标签生成一个txt的文件，文本第一列为文件名，第二列为标签号，中间用空格隔开。
提示：使用时更改path的文件地址，j是自定义的标签数量。

import os
import sys

path='D:/0_Data/2021-4-20(normal)/shipintu4/x_test'
filenames=os.listdir(path) #读取path内所有文件名返回列表
i=0
j=0

with open('D:/0_Data/2021-4-20(normal)/shipintu4/y_test.txt', "w") as f:
        for filename in filenames:
                print(filename+','+str(i))
                f.write(filename+' '+str(i)+'\n')
                j=j+1
                if j==12:
                        i=i+1
                        j=0

---