修改下面代码去掉排序部分,要将给定路径下的所有文件名写入一个文本文件import os dir=r'C:\Users\huawei\Desktop\pythonProject1\data\train' fp = open('./train_list.txt','w+') imgfile_list = os.listdir(r'C:\Users\huawei\Desktop\pythonProject1\data\train') imgfile_list.sort(key= lambda x:int(x[:])) #print(img_list) seqsize =17 for imgfile in imgfile_list: filepath = os.path.join(dir,imgfile) img_list = os.listdir(filepath) img_list.sort(key=lambda x: int(x[:-4])) #滑窗取序列,步长为8 for i in range(0, len(img_list)-seqsize, 8): for j in range(i,i+seqsize): img = img_list[j] path = os.path.join(filepath, img) if j == i+seqsize-1: fp.write(path+'\n') else: fp.write(path+' ') fp.close()
时间: 2023-06-09 09:05:42 浏览: 43
将下面的代码修改一下,去掉排序部分,将给定路径下所有的文件名写入一个文本文件中:
```python
import os
dir = r'C:\Users\huawei\Desktop\pythonProject1\data\train'
fp = open('./train_list.txt', 'w')
imgfile_list = os.listdir(dir)
for imgfile in imgfile_list:
fp.write(imgfile + '\n')
fp.close()
```
这段代码将给定路径下的所有文件名写入了一个名为"train_list.txt"的文本文件中。