python - 从 os.walk 创建字典理解

Question

我正在尝试创建所有 mp3 文件的字典，其中键是 mp3 文件名，值是该文件的文件路径。

我愿意:

for root, dirs, files in os.walk(rootDir, topdown='true'):
    source_files_now = {filename:root for filename in files if filename[-4] == '.mp3'}
    print(source_files_now);

但是字典 source_files_now 最终是空的。

目录中肯定有mp3文件。

有什么想法吗？

Answer 1

filename[-4] == '.mp3' 检查倒数第三个字符是否为 '.mp3'。这真的没有意义。

只需使用.endswith():

filename.endswith('.mp3')

如果您想修复您的代码，请将字符串切片，以便切片包含其他字符:

filename[-4:] == '.mp3'

在比较之前，我还会将文件名转换为小写。您最终可能会跳过全部大写的文件名。