python - 如何使用 Python 分隔文件名的数字部分？

Question

我正在尝试编写一个简短的程序，它可以查看目录，获取图像文件的文件名，并将它们附加到与目录名称相匹配的位置，并对它们进行重新编号和排序以便稍后处理。到目前为止，我可以获取文件夹的名称，并使用以下命令替换文件名的特定部分；

import os

print os.getcwd()
str = os.getcwd()
ext =  str.split("/")[-1]
print ext

separ = os.sep
folder = str
for n in os.listdir(folder):
    print n
    if os.path.isfile(folder + separ + n):
        filename_zero, extension = os.path.splitext(n)
        os.rename(folder + separ + n , folder + separ + filename_zero.replace('image',ext) + extension)

for n in os.listdir(folder):
    print n

我不能做的是单独获取数字部分。我的文件名类型为storm000045.tiff，并且从来没有下划线或点供我分隔它们。任何建议表示赞赏。提前致谢!

Answer 1

使用这个简单的函数:

import re
def get_name_and_number(text):
    return re.match(r'(\D+)(\d+).*', text).groups()

示例:

>>> get_name_and_number('storm000045.tiff')
('storm', '000045')

<小时/>

或者这个:

def extract_numbers(text):
    return ''.join([x for x in text if x.isdigit()])

示例:

>>> extract_numbers('storm000045.tiff')
'000045'