将目录设置为脚本路径

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

homedir = '.'

或者

import os
homedir = os.getcwd()

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

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

Answer 2

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

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

输出：

.\x.py
c:\test