Python 列表理解不适用于函数参数

Question

我正在尝试获取目录中所有文件的列表并删除它们。我使用了以下代码并且运行良好。

代码:

import os

file_list=os.listdir(mypath)
for x in file_list:
    os.remove(mypath+"/"+x)

但是当我尝试使用列表推导方式来执行此操作时，它给出了一个语法错误。

import os

file_list=os.listdir(mypath)
os.remove(mypath+"/"+x) for x in file_list

我们不能用列表推导中的参数调用一个函数，还是有其他方法？

Answer 1

import os

file_list=os.listdir(mypath)
[os.remove(mypath + os.sep +x) for x in file_list]

或者只是

[os.remove(mypath + os.sep +x) for x in os.listdir(mypath)]

或

for x in os.listdir(mypath):
    os.remove(mypath + os.sep +x)