將目錄設置爲腳本路徑

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