按索引/编号选择目录中的文件

Question

我需要从列表中读取文件。具体来说，我需要查找存储在特定目录中的文件。由于它可能不记得存储在那里的所有文件或只是正确的名称，我需要列出他们的。为了完成这一步，我发现了这篇有用的帖子：

How do I display the the index of a list element in Python?

现在我有一个如下列表：

file=[]

for f in glob.glob("*.txt"):
    file.append(f)


for (num,item) in enumerate(file):
    print(num+1,item)        # updated after Thierry's comment

输出：

1 file 1.txt
2 file_2.txt
3 file_3.txt
...

我想使用

读取文件

query=input("Please add your selection: ") # just the number
if int(query) in enumerate(datasets):
    dataset=pd.read_csv('path_name'+1) # I would need to add the path to the file and, by the number, the filename 
df = pd.DataFrame(dataset)

它给了我错误：TypeError: 'module' object is not iterable.

我该怎么做？

Answer 1

就像 Thierry 在 cmets 中所说的那样，我可能会避免使用这种方法并尝试这样的方法：

files = [f for f in glob.glob("*.txt")]

for fi, f in enumerate(files):
    print(fi, f)

query = input("Please add your selection: ") # just the number
df = pd.read_csv(files[int(query)])

Answer 2

您也可以尝试使用字典稍有不同的方法。

import os
import pandas as pd

# iterate over a list of files in directory
filelist = [files for files in os.listdir("<path to directory>")]

# convert the files list to a dictionary
filelist_dict = { ind: name for (ind,name) in enumerate(filelist) }

# ask for user input
userinput = int(input("Enter a number :"))

# open the file based on the key input by user
with open(filelist_dict[userinput]) as file:
       df = pd.read_csv(file)  # read the csv file into a pandas dataframe
       print(df)               # print the dataframe

希望这会有所帮助:)