按特定顺序重命名具有两个不同名称的图像

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'))

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