如何重命名google colab文件夹中的所有文件

from google.colab import drive
drive.mount('/content/drive')

import os
os.chdir('/content/drive/MyDrive/your_folder_name')

import os

folder_path = '/content/drive/MyDrive/your_folder_name'  # 文件夹路径
new_file_name = 'new_file_name'  # 新文件名

# 遍历文件夹中的所有文件
for file_name in os.listdir(folder_path):
    file_path = os.path.join(folder_path, file_name)
    
    # 构造新的文件名
    new_file_path = os.path.join(folder_path, new_file_name + os.path.splitext(file_name)[1])
    
    # 重命名文件
    os.rename(file_path, new_file_path)