Question

我有一个包含numpy文件的文件夹。我需要创建一个numpy列表，我将放置所有numpy文件名。为此，我使用此代码：

import os

a = open('C:\\Users\\user\\My_Test_Traces\\Traces.list_npy', "w")
for path, subdirs, files in os.walk(r'C:\\Users\\user\\CPA_test_1000_Tests\\test'):
   for filename in files:
      f = os.path.join(path, filename)
      a.write(str(f) + os.linesep)

结果，它为我提供了我的文件列表及其路径：

   C:\\Users\\user\\My_Test_Traces\\ Trace1_Pltx1

   C:\\Users\\user\\My_Test_Traces\\ Trace2_Pltx2

   C:\\Users\\user\\My_Test_Traces\\ Trace3_Pltx3

   C:\\Users\\user\\My_Test_Traces\\ Trace4_Pltx4

但是，我只需要文件名，行之间不需要任何空格：

   Trace1_Pltx1
   Trace2_Pltx2
   Trace3_Pltx3
   Trace4_Pltx4

Answer 1

有一个函数tyhat正在做你需要的东西，你应该试试这个：

os.path.basename(path)

Answer 2

你可以使用，

from os.path import basename
print basename('path/file.txt')

如果您不想要扩展程序，请参阅post

Answer 3

也许这更好：

import os

folder = 'C:\\Users\\user\\CPA_test_1000_Tests\\test'

with open('C:\\Users\\user\\My_Test_Traces\\Traces.list_npy', 'w') as fp:
    fp.write(os.linesep.join(os.listdir(folder)))

如何创建包含在同一文件夹中的文件的文件列表？

3 个答案: