python - 如何在使用 os.listdir 时删除 .py 扩展名

Question

在我的代码中，我想对目录中存在的所有文件执行导入文件名。我在工作区下有一个文件abc.py。我目前正在做以下事情:

for filename in os.listdir(homedir/workspace)
    exec "import " + filename
    filename = eval(filename + '.function(variable)')

问题是，它没有执行 import abc，而是执行 import abc.py，然后显示错误 no module named py

我该如何解决？

提前致谢!

Answer 1

您可以使用 os.path.splitext

os.path.splitext(filename)[0]

在返回的双元素数组中，第一个元素只是文件名，第二个元素是扩展名。为了更加安全，您可以仔细检查您是否确实获取了一个 .py 文件:

if os.path.splitext(filename)[1] == '.py':
    # do stuff