Python:如何在使用 os 重命名文件时保留文件扩展名？

Question

假设我有一个包含 n 个 csv 文件的文件夹，我想重命名这些文件。新文件名将类似于 ABxxxx，其中 xxxx 是一个从 1 到 1000 的递增数字。

在执行此操作时，如何保留原始文件扩展名，即 csv？

到目前为止，我所做的只是更改了文件名，但删除了扩展名:

directory=r'C:\Me\MyDir'
subdir=[x[0] for x in os.walk(directory)]
subdir.pop(0)

for i in subdir:
    temp_dir=r''+i
    os.chdir(temp_dir)
    a='A'
    b='B'
    for file in glob.glob("*.csv"):
        for i in range(1,1001):
           newname=a+b+i
        os.rename(file,newname)

Answer 1

您可以简单地将 '.csv' 附加到您的新文件名:

os.rename(file, newname + '.csv')

通常(对于任何文件类型)，更好的方法是先使用 os.path.splitext 获取现有扩展名然后将其附加到新文件名。

oldext = os.path.splitext(file)[1]
os.rename(file, newname + oldext)