以特定顺序使用两个不同的名称重命名图像

Question

我需要使用Python重命名文件夹中的992个图像名称。图像的名称应根据其顺序更改。例如

 old: image_1     new: P1_ES_1
 old: image_2     new: P1_ES_2
 old: image_3     new: P1_ES_3
 old: image_4     new: P1_ED_1
 old: image_5     new: P1_ED_2
 old: image_6     new: P1_ED_3
 old: image_7     new: P2_ES_1
 old: image_8     new: P2_ES_2
 old: image_9     new: P2_ES_3
 old: image_10    new: P2_ED_1

...

这是@anki提供的与我稍有更改的代码段，但是问题是新名称以ED开头，但应该是ES。任何帮助将不胜感激。

import os
import glob

path = 'F:/my_data/imagesResized/'
#path = 'F:/my_data/labelsResized/'

fns = glob.glob(path + '*.png')
fns.sort(key = len)
print(fns)

es_or_ed = 'ES'

for i, fn in enumerate(fns):

    # Check for ED or ES

    if i % 3 == 0 and es_or_ed == 'ES':
        es_or_ed = 'ED'

    elif i % 3 == 0 and es_or_ed == 'ED':
        es_or_ed = 'ES'

    # Create new filename
    new_fn = 'P{}_{}_{}'.format(i // 6 + 1, es_or_ed, i%3+1) 
    #new_fn = 'P{}_{}_{}_{}'.format(i // 6 + 1, es_or_ed, i%3+1,"label")

    # rename...S
    os.rename(fn, os.path.join(path, new_fn + '.png'))

Answer 1

当前以ES开头的原因是零索引。在第一次循环迭代期间i==0时，您的代码将es_or_ed的值更改为ED。

我修改了您的代码以解决此问题，并执行了正确的原始文件名排序，因为看起来文件名中没有前导零，并且您希望10在9之后而不是1之后出现。是我从此answer借用的一个函数，它将对文件名列表进行正确排序。

import os
import glob

import re

def natural_sort_key(s, _nsre=re.compile('([0-9]+)')):
    return [int(text) if text.isdigit() else text.lower()
            for text in _nsre.split(s)]   

path = 'F:/my_data/labelsResized/'

fns = glob.glob(path + '*.png')

es_or_ed = 'ED'

for i, fn in enumerate(sorted(fns, key=natural_sort_key)):

    # Check for ED or ES

    if (i+1) % 3 == 1 and es_or_ed == 'ED':
        es_or_ed = 'ES'

    elif (i+1) % 3 == 1 and es_or_ed == 'ES':
        es_or_ed = 'ED'

    # Create new filename
    new_fn = 'P{}_{}_{}'.format(i // 6 + 1, es_or_ed, i%3+1) 

    # rename...S
    os.rename(fn, os.path.join(path, new_fn + '.png'))

结果（来自将原始文件名附加到新文件名的代码（未显示））：