Python批量修改文件后缀名

直接上代码
将需要修改的文件和代码放在一个目录下

import os


def file_extension_change():

    files = os.listdir('.')
    for filename in files:
        extension = os.path.splitext(filename)
        # 如果后缀是txt，就改成.ipynb
        if extension[1] == '.txt':
            newname = extension[0] + '.ipynb'
            os.rename(filename, newname)


if __name__ == '__main__':
    file_extension_change()