为什么我的 for 循环在 Python 中无法正常工作？

Question

Python 中的新功能，希望你能帮助

我有一个带有这个 for 循环的脚本，用于检查目录中的文件列表，我需要做的是，如果每个文件以“xlsx”结尾，它会执行一段代码，如果它以“csv”结尾，它执行其他块代码。 我必须创建变量 extension 来提取文件的扩展名，因为如果我

for file in file_list:
        #to extract the file extension:
        extension = os.path.splitext(file)[1][1:]
        print(file)

        if extension.endswith('xlsx'):
            all_df_list.append(pd.read_excel(file, header=0, engine='openpyxl'))
            a = round(time.time() - start_time,2)
            print(a)
        elif extension.endswith('csv'):
            all_df_list.append(pd.read_csv(file, chunksize=10000))
            a = round(time.time() - start_time,2)
            print(a)
        else:
            print("Not working")

运行时 使用 file ，它会显示错误 WindowsPath' object has no attribute 'endswith'在整个脚本中，它启动循环并执行第一个文件的命令，但随后一直执行到 else 语句。它应该对每个文件执行相同的操作...这些是结果（放置 xxx 来隐藏敏感信息）

Z:\DataFiles\XXX\XXX\XXX\XXX\XXX\XXX_XXX_0000.xlsx
2.43
Z:\DataFiles\XXX\XXX\XXX\XXX\XXX\XXX_XXX_2019.XLSX
Not working
Z:\DataFiles\XXX\XXX\XXX\XXX\XXX\XXX_XXX_2020.XLSX
Not working
Z:\DataFiles\XXX\XXX\XXX\XXX\XXX\XXX_XXX_2021.XLSX
Not working
Z:\DataFiles\XXX\XXX\XXX\XXX\XXX\XXXX_2022.XLSX
Not working

你能指出我做错了什么吗？

Answer 1

由于 file 是一个 WindowsPath 对象（如错误所示），您可能需要使用 .suffix，并使用 .lower() 将其小写与小写扩展名比较：

for file in file_list:
    if file.suffix.lower() == '.xlsx':
        all_df_list.append(pd.read_excel(file, header=0, engine='openpyxl'))
    elif file.suffix.lower() == '.csv':
        all_df_list.append(pd.read_csv(file, chunksize=10000))
    else:
        print("Not working")
        continue
    print(round(time.time() - start_time, 2))

Answer 2

您的脚本正在查找 .xlsx 和 .csv，因此 .XLSX 在 else 中进行处理：
块。
当您将值分配给变量扩展名时，可以使用函数.lower()。

extension = os.path.splitext(file)[1][1:].lower()

Answer 3

我尝试了不同的解决方案，按照建议，我只是使用了：

        if file.suffix.lower() == '.xlsx':
            all_df_list.append(pd.read_excel(file, header=0, engine='openpyxl'))
            a = round(time.time() - start_time,2)
            print(a)
        elif file.suffix.lower() == '.csv':
            all_df_list.append(pd.read_csv(file, chunksize=10000))
            a = round(time.time() - start_time,2)
            print(a)

并且它有效。多谢！！！

Answer 4

不能在循环外定义 extension = os.path.splitext(file)[1][1:] 吗？

Answer 5

使用 pathlib for Pathing 您可以得到字符串来自.name！

>>> import pathlib
>>> p = pathlib.Path("foo/bar/baz.EXT")
>>> p.name.lower().endswith(".ext")
True