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

1. 如何获取文件扩展名？

文件扩展名是指文件名中最后一个“.”字符后面的部分，它通常用于表示文件类型。在Python中，我们可以通过对文件名进行字符串操作来获取文件扩展名。

1.1 使用split()函数分割文件名

Python的字符串类型有一个内置函数split()，可以按照指定的分隔符将字符串分割成多个子字符串，并返回一个列表。我们可以使用split()函数来截取文件名中的扩展名部分。

filename = 'example.txt'

extension = filename.split('.')[-1]
print(extension)  # 输出：'txt'

上面的代码中，我们首先定义了一个文件名变量filename，然后使用split()函数将其按照“.”字符进行分割，从而得到一个列表。因为扩展名在文件名的最后一个“.”字符后面，所以我们使用列表的最后一个元素来获取扩展名。

1.2 使用os模块中的splitext()函数

Python的os模块中提供了一个splitext()函数，可以直接从文件名中获取扩展名。

import os

filename = 'example.txt'
extension = os.path.splitext(filename)[-1]
print(extension)  # 输出：'.txt'

上面的代码中，我们首先导入了os模块，然后使用splitext()函数将文件名分隔成两部分，一部分是文件名，另一部分是扩展名。同样地，我们使用列表的最后一个元素来获取扩展名。

2. 扩展名的应用

文件扩展名通常用于表达文件类型，而不同的文件类型需要使用不同的程序来打开。因此，在读取或保存文件时，正确地获取文件扩展名非常重要。

2.1 读取不同类型的文件

在Python中，不同类型的文件通常需要使用不同的读取方法。我们可以根据读取的文件扩展名来确定所需的读取方法。

filename = 'example.csv'

extension = os.path.splitext(filename)[-1]
if extension == '.csv':
    # 使用csv模块读取csv文件
    pass
elif extension == '.txt':
    # 使用open()函数读取文本文件
    pass
else:
    raise TypeError('Unsupported file type: {}'.format(extension))

上面的代码中，我们首先获取文件名的扩展名，并根据扩展名选择不同的读取方法。如果文件扩展名是.csv，我们就使用csv模块来读取文件；如果是.txt，我们则使用open()函数来打开文件。

2.2 保存不同类型的文件

与读取文件类似，在保存文件时，也需要根据文件类型选择不同的保存方法。同样地，我们可以使用文件扩展名来确定所需的保存方法。

filename = 'example.csv'

extension = os.path.splitext(filename)[-1]
if extension == '.csv':
    # 使用csv模块保存数据到csv文件
    pass
elif extension == '.txt':
    # 使用open()函数保存数据到文本文件
    pass
else:
    raise TypeError('Unsupported file type: {}'.format(extension))

上面的代码中，我们首先获取文件名的扩展名，并根据扩展名选择不同的保存方法。如果文件扩展名是.csv，我们就使用csv模块来保存数据到文件；如果是.txt，我们则使用open()函数来保存数据。

3. 总结

在Python中，获取文件扩展名是一个常见的操作。我们可以使用字符串操作和os模块中的函数来获取文件扩展名，然后根据文件类型来选择不同的处理方法。