欢迎访问宙启技术站
智能推送

os.path模块获取文件扩展名的方法

发布时间:2024-01-20 11:41:38

os.path模块是Python自带的模块之一,提供了一些处理路径的函数。其中,获取文件扩展名的方法是os.path.splitext()。

os.path.splitext()方法接受一个文件名作为参数,返回一个元组,其中包含文件名和文件扩展名。具体用法如下:

import os

# 获取文件扩展名
filename = "example.txt"
extension = os.path.splitext(filename)[1]

print("文件名:", filename)
print("文件扩展名:", extension)

以上代码输出的结果为:

文件名: example.txt
文件扩展名: .txt

在上述例子中,我们首先引入了os模块。然后,定义了一个变量filename,值为"example.txt",这是一个文件名。接下来,我们调用os.path.splitext(filename)方法,它会返回一个元组,其中包含了文件名和文件扩展名。最后,我们通过下标[1]获取了文件扩展名,并将结果赋值给变量extension。最后,我们打印了文件名和文件扩展名。

需要注意的是,os.path.splitext()方法获取的文件扩展名前面会包含一个点号。所以,在上述例子中,文件扩展名是".txt",而不是"txt"。如果要去除这个点号,可以使用字符串的strip()方法,如下所示:

import os

# 获取文件扩展名,去除点号
filename = "example.txt"
extension = os.path.splitext(filename)[1].strip('.')

print("文件名:", filename)
print("文件扩展名:", extension)

以上代码输出的结果为:

文件名: example.txt
文件扩展名: txt

在上述例子中,我们在os.path.splitext(filename)[1]之后调用了strip('.')方法,将文件扩展名中的点号去除了。这样,我们得到的文件扩展名就是"txt",而不是".txt"。