如何在Python中获取文件扩展名？

在Python中，可以使用os.path模块来获取文件的扩展名。os.path模块提供了一系列有用的函数，用于处理文件和路径。

要获取文件的扩展名，我们可以使用os.path.splitext函数。该函数将文件名分成文件名部分和扩展名部分，并将其作为元组返回。下面是使用os.path.splitext函数获取文件扩展名的示例代码：

import os

filename = 'example.csv'
extension = os.path.splitext(filename)[1]

print('文件名：', filename)
print('扩展名：', extension)

输出：

文件名： example.csv
扩展名： .csv

从输出中可以看出，os.path.splitext函数返回的是带有点号的扩展名。如果希望去掉点号，可以使用字符串的lstrip方法。下面是去掉点号的示例代码：

import os

filename = 'example.csv'
extension = os.path.splitext(filename)[1].lstrip('.')

print('文件名：', filename)
print('扩展名：', extension)

输出：

文件名： example.csv
扩展名： csv

还可以使用字符串的split方法来获取文件名和扩展名部分。下面是使用split函数获取文件扩展名的示例代码：

import os

filename = 'example.csv'
name, extension = filename.split('.')

print('文件名：', name)
print('扩展名：', extension)

输出：

文件名： example
扩展名： csv

不过，split函数在文件名中包含多个点号的情况下会出错。因此，推荐使用os.path.splitext函数获取文件扩展名。

以上就是在Python中获取文件扩展名的方法。使用os.path模块提供的函数，我们可以轻松地获取文件的路径、文件名、扩展名等信息。