指定したファイル名の拡張子のデータ構造に共通するデザインと機能文字列バック

指定したファイル名の拡張子に戻るには、機能を設計

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