Làm cách nào để trích xuất dữ liệu từ một chuỗi với Biểu thức chính quy Python?

Đoạn mã sau trích xuất dữ liệu như first_id, second_id, danh mục từ các chuỗi đã cho

Ví dụ

import re
s = 'TS001B01.JPG'
match = re.match(r'(TS\d+)([A|B])(\d+)\.JPG', s)
first_id = match.group(1)
category = match.group(2)
second_id = match.group(3)
print first_id
print category
print second_id

Đầu ra

Điều này cho ra kết quả

TS001
B
01