python - 使用python中的os将文件名放入数组中

Question

我想创建一个程序，该程序将使用 os 来创建目录的 ls ，然后将文件放入数组中。然而，ls 并没有一一列出名称，所以我有点卡住了，因为没有任何常数可以让我知道下一个名称何时开始。

这是实际的代码:

import os
cd = input("Change directory to:   ")

while cd != "x":
    os.chdir(cd)
    command = os.popen('ls')
    ls = command.read()
    print(ls)
    cd = input("Change directory to:   ")
    count = 0
    word = ""

for character in ls:
    while count == 0 :
        if character!= " ":
            count = 1
            word += character

        elif character == " ":
            print(word)

因此，当程序启动时，它应该询问您要更改到哪个目录，然后在该目录上运行 ls 。然后，如果您输入 x，它应该只将第一个目录打印到屏幕上。然而，正如我所说，它只是打印整个列表。

有人可以帮我解决这个问题吗？

任何帮助表示赞赏。谢谢!

Answer 1

您可以使用已经准备好的os.listdir - 它将为您提供目录中的所有内容 - 文件和目录。

但是如果您只想要文件，您可以添加 os.path 的用法:

from os import listdir
from os.path import isfile, join
files = [file for file in listdir(path) if isfile(join(path, file))]