import csv
# 遍历文件夹
file = os.listdir('/data2/nlp/SwinBERT/datasets/MSVD/videos')
f = open('文件名.csv','w', encoding='utf-8', newline='')
csv_writer = csv.writer(f)
#写入数据
for i,j in enumerate(file):
csv_writer.writerow([i,j])
f.close() #很重要,不关闭文件可能后续读取数据不全
f = open('文件名.csv',newline='')
ff = csv.reader(f) #读取数据
for data in ff:
i,j = data
print(i,j)
结果
1954 EiylMb_mWk4_2_20.avi
1955 SKhmFSV-XB0_12_18.avi
1956 ACOmKiJDkA4_47_54.avi
1957 2F9jX7u5nOg_83_97.avi
1958 kWLNZzuo3do_31_37.avi
1959 4PcL6-mjRNk_11_18.avi
1960 504gE1DA_4g_1_6.avi
1961 Ugb_uH72d0I_8_17.avi
1962 nlU3crMsbWI_19_23.avi
1963 oSnYUfCGiCA_7_17.avi
1964 AzMrltkXGJk_0_10.avi
1965 fnpp8v9NbmY_181_188.avi
1966 pRpeEdMmmQ0_1_18.avi
1967 0hyZ__3YhZc_632_637.avi
1968 a1Y73sPHKxw_0_5.avi
1969 uiLr9bdOL0M_23_30.avi