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)