字符串和常用的数据结构之设计一个函数返回给定文件名的后缀名

设计一个函数返回给定文件名的后缀名

def get_suffix(filename, has_dot=False):

 """
 获取文件名的后缀名

 :param filename:文件名
 :param has_dot:返回的后缀名是否需要带点
 :return:文件的后缀名
 """


 pos = filename.rfind('.')
 if 0 < pos < len(filename) - 1:
  index = pos if has_dot else pos + 1
  return filename[index:]
 else:
  return ''

测试

print(get_suffix('aa'))
print(get_suffix('aa.txt'))
print(get_suffix('aa.py'))
print(get_suffix('aa.txt.py'))

来源：CSDN

作者：匡夆

链接：https://blog.csdn.net/weixin_46108954/article/details/104637432