Python习题 097：提取文件名中的扩展名

说明

《Python 教程》持续更新中，提供建议、纠错、催更等加作者微信: gairuo123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

（编码题）以下字符串是一个文件名，用 Python 编写代码，提取出它的扩展名。

data = 'beijing.mp4'

输出: mp4

参考答案

代码如下：

data = 'beijing.mp4'

data.split('.')
# ['beijing', 'mp4']

data.split('.')[-1]
# 'mp4'

可以利用 pathlib 模块：

from pathlib import Path

data = 'beijing.mp4'
file = Path(data)

file.suffix[1:]
# mp4

用字符串的 split() 方法，按符号拆开字符串，形成一个列表，列表的最后一个元素就是展示名，用索引的方法获取即可。

（完）