Python語言非常適合文件處理和數(shù)據(jù)分析,因此我們經(jīng)常需要獲得文件的擴展名以判斷文件類型。Python提供了方便的操作,讓我們輕松獲得文件的擴展名。
import os def get_ext(filepath): filename, extension = os.path.splitext(filepath) return extension print(get_ext('example.pdf'))
在這個例子中,我們使用了os模塊提供的splitext()函數(shù),該函數(shù)可以將文件路徑拆分為文件名和擴展名。然后我們可以直接返回擴展名extension,以便后續(xù)判斷。
另外,我們也可以通過字符串操作來獲取文件的擴展名。例如:
def get_ext(filepath): return filepath.split('.')[-1] print(get_ext('example.pdf'))
使用split()函數(shù)將文件路徑按照'.'進行分割,并取得最后一個字符串即可獲得擴展名。但是這種方法在文件名中出現(xiàn)多個'.'的情況下可能會出現(xiàn)錯誤,因此需要考慮文件名中'.'的情況。
以上就是Python中獲取文件擴展名的兩種方法。使用os模塊的splitext()函數(shù)可以更安全地獲取擴展名。在實際應(yīng)用中,我們可以根據(jù)自己的需求靈活地選擇兩種方法中的其中一種或組合使用。