Python 从列表中选择一个文件

Question

我有一个文件夹，其中包含几个我将用 python 解析的日志文件。

我会显示文件夹中包含的文件列表，例如:

• [1] 文件名1.log
• [2] 文件名2.log

然后用户可以选择正确的文件并写入文件列表号。

例如，要解析文件“FileName2.log”，用户按 2。

在我的脚本中，我可以显示文件列表，但我现在不知道如何按索引从列表中选取文件。

这是我的脚本

import os
import sys

items = os.listdir("D:/Logs")

fileList = []

for names in items:
    if names.endswith(".log"):
        fileList.append(names)

cnt = 0
for fileName in fileList:
    sys.stdout.write( "[%d] %s\n\r" %(cnt, fileName) )
    cnt = cnt + 1


fileName = raw_input("\n\rSelect log file [0 -" + str(cnt) + " ]: ")

感谢您的帮助!

Answer 1

import os
import sys

items = os.listdir("D:/Logs")

fileList = [name for name in items if name.endswith(".log")]

for cnt, fileName in enumerate(fileList, 1):
    sys.stdout.write("[%d] %s\n\r" % (cnt, fileName))

choice = int(input("Select log file[1-%s]: " % cnt))
print(fileList[choice])

您拥有经过少量修改的代码版本，希望这能解决您的目的