Làm cách nào để tìm một số trong chuỗi bằng Python?

Question

Tôi có một tệp được gọi là FILE-1.txt hoặc FILE-340.txt. Tôi muốn có thể lấy số từ tên tập tin. Tôi đã thấy rằng tôi có thể sử dụng

numbers = re.findall(r'\d+', '%s' %(filename)) 

để có được một danh sách có chứa số lượng, và sử dụng các số [0] để có được số lượng chính nó như là một chuỗi ... Nhưng nếu tôi biết nó chỉ là một trong số, có vẻ như vòng xoay và không cần thiết làm cho một danh sách để có được nó. Có một cách khác để làm điều này?

---

Chỉnh sửa: Cảm ơn! Tôi đoán bây giờ tôi có một câu hỏi khác. Thay vì nhận được một chuỗi, làm thế nào để có được số nguyên?

Answer 1

Sử dụng search thay vì findall:

number = re.search(r'\d+', filename).group() 

Hoặc:

number = filter(str.isdigit, filename) 

Answer 2

Nếu bạn muốn chương trình của bạn có hiệu quả

sử dụng này:

num = filename.split("-")[1][:-4] 

này sẽ chỉ làm việc với các ví dụ mà bạn thấy

Answer 3

Một cách khác chỉ để cho vui:

In [1]: fn = 'file-340.txt' 

In [2]: ''.join(x for x in fn if x.isdigit()) 
Out[2]: '340' 

Answer 4

Thêm vào comment FJ, nếu bạn muốn một int, bạn có thể sử dụng:

numbers = int(re.search(r'\d+', filename).group()) 

Answer 5

đáp lại câu hỏi mới của bạn, bạn có thể cast string một int:

>>>int('123') 
123