개발 꿀팁/PYTHON

Python 파일 이름에서 숫자 읽기

Jammie 2022. 11. 28. 13:01
반응형

정규 표현을 사용할 수 있습니다

regex = re.compile(r'\d+')

그런 다음 일치하는 문자열을 가져옵니다

regex.findall(filename)

숫자가 포함된 문자열 목록을 반환합니다.만약 당신이 실제로 정수를 필요로 한다면, int:를 사용할 수 있습니다

[int(x) for x in regex.findall(filename)]

각 파일 이름에 숫자가 하나만 있는 경우 regex.search(filename).group(0)을 사용할 수 있습니다.일치하는 항목이 없으면 위의 행은 NoneType에 attribute Group이 없음을 나타내는 attribute Error를 생성합니다

import re
 
name = 'CP_epoch164.pth'
regex = re.compile(r'\d+')
num = int(max(regex.findall(name)))
print(num)

주의: regex.findall(name) 반환 값은 list이므로 max 함수로 값을 취하고 반올림합니다

 

반응형

'개발 꿀팁 > PYTHON' 카테고리의 다른 글

파이썬 상대 경로  (0) 2022.11.28
Pytorch이상 처리  (1) 2022.11.28
Python에서 logging 기본 용법  (0) 2022.11.28
Python 폴더 이동  (0) 2022.11.28
Python_ 객체 지향  (0) 2022.11.25