python - 如何通过将列表元素与Python中的字符串进行比较来找到列表元素的索引？

Question

假设我有一个包含一些文件的列表，如下所示:

file_list = ['folder\\text.txt', 'folder\\data.csv', 'folder\\picture.png']

并且还声明了一个字符串:

file = 'text.txt'  #Did not include 'folder\\', intentionally.

如何使用“file”变量获取 file_list 的索引？

我尝试过以下方法:

index = file_list.index(file)
filename = file_list[index]
print(filename)

但是，我得到了这个结果:

ValueError: 'text.txt' is not in list

我已经明白为什么我的解决方案失败了，但是有没有一种有效的解决方案可以通过使用字符串变量来获取索引？

Answer 1

您应该检查file_list的每个元素以找出file的索引:

def find_index(file, file_list):
    for idx, file_name in enumerate(file_list):
        if file in file_name:
            print(f"idx:{idx}, file_name:{file_name}")
            return idx


if __name__ == '__main__':
    file_list = ['folder\\text.txt', 'folder\\data.csv', 'folder\\picture.png']
    file = 'text.txt'
    idx = find_index(file, file_list)
    if idx:
        print(file_list[idx])
    else:
        print(f"Couldn't find idx of {file} ")

输出:

idx:0, file_name:folder\text.txt
folder\text.txt