Question

我想要做的是创建一个设置为主文件夹脚本的变量homedir。我该怎么做呢？就像说脚本位于C：\ blah \，我希望homedir被分配给它。

homedir = (current directory)
pdfList = []

def getfiles():
    directory = os.listdir(homedir)
    for file in directory:
        if "pdf" in file:
            pdfList.append(file)

Answer 1

__file__模块属性包含脚本位置，但它可能是相对的。以下文件为c:\test\x.py。

import os
print(__file__)
homedir = os.path.abspath(os.path.dirname(__file__))
print(homedir)

输出：

.\x.py
c:\test

Answer 2

homedir = '.'

或

import os
homedir = os.getcwd()

将导致os.listdir(homedir)列出当前工作目录（即执行脚本的目录）中的文件（和子目录）。请注意， NOT 必须与包含脚本的目录相同。对于该目录，请使用

import os
homedir = os.path.dirname(__file__)

将目录设置为脚本路径

2 个答案: