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

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

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'))