[Python] 문자열에서 숫자 추출

1. re.sub()

•  모든 숫자들을 1개의 문자열로 추출
• sub()는 String에서 pattern과 일치하는 문자들을 repl로 교체

re.sub(pattern, repl, string)

 

 

import re

string = 'abc1234abc'
num = re.sub(r'[^0-9]', '', string)
print(num)

# 1234

 

 

 

2. re.findall()

- r'\d+'

• string에서 pattern에 해당하는 내용들을 찾아서 리스트로 리턴
• r'\d+'는 1회 이상 반복되는 숫자들에 대한 패턴을 의미
• 연속된 숫자들을 하나의 Integer로 추출

string = 'aab 1234, 543 bbc'
num = re.findall(r'\d+', string)
print(num)

# ['1234', '543']

 

- r'\d'

• 1개의 숫자
• 1개 이상의 반복을 의미하는 +가 제외된 패턴으로 패턴을 찾으면, 숫자 1개 단위로 추출되어 리스트에 저장

string = 'aab 1234, 543 bbc'
num = re.findall(r'\d', string)
print(num)

# ['1', '2', '3', '4', '5', '4', '3']