Python某个文件夹下的所有文件的重命名

flag为1文件名前+Python_
flag为2删除Python_

import os
flag = 2
# 获取当前目录下所有文件名并返回一个列表
filelist = os.listdir()
# 遍历列表进行重命名
for i in filelist:
    if flag == 1:
        new_name = 'Python_' + i
    elif flag == 2:
        length = len('Python_')
        new_name = i[length:]
    os.rename(i,new_name)