Python_目錄操作_os.path.splitext()

os.path.splitext(path)

說明：將對應路徑的文件名和后綴名分割

示例：

>>> #定義參數內容
... import os
>>> path1='E:\test\6.txt'#文件路徑
>>> path2='E:\test'#目錄
>>>
>>> #用splitext()方法切割
... split_path1=os.path.splitext(path1)
>>> split_path2=os.path.splitext(path2)
>>>
>>> #打印結果
... print(split_path1)#正常切割
('E:\test\x06', '.txt')
>>> print(split_path2)#目錄切割后異常
('E:\test', '')
>>>

 

簡單運用：統計整個e盤下，txt文件的總數

>>> file_number=0
>>> for root,dirs,files in os.walk("e:\\"):
...     for file in files:
...         if os.path.splitext(file)[1]==".txt":#將對應的文件與文件名分割
...             file_number+=1
...             #print (file)
...
>>>
>>> print (file_number)
246

其中os.walk()的簡單說明，請參照：python中：os.walk()的簡單說明

 

上面的方法等價於：

>>> for root,dir_name,file_name in os.walk(r"e:"):
...     for file in file_name:
...         if ".txt" in file:
...             count_txt += 1
...
>>> print(count_txt)
246