Làm cách nào để trích xuất các số từ tên tệp bằng Python?

Question

tôi cần phải chỉ trích các số từ tên tập tin như:

GapPoints1.shp

GapPoints23.shp

GapPoints109.shp

Làm thế nào tôi có thể chỉ trích các số từ những tập tin này sử dụng Python? Tôi sẽ cần phải kết hợp điều này vào một vòng lặp for.

Answer 1

bạn có thể sử dụng biểu thức thông thường:

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

Sau đó, để có được những chuỗi phù hợp:

regex.findall(filename) 

này sẽ trả về một danh sách các chuỗi có chứa các con số. Nếu bạn thực sự muốn số nguyên, bạn có thể sử dụng int:

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

Nếu chỉ có 1 số trong mỗi tên tập tin, bạn có thể sử dụng regex.search(filename).group(0) (nếu bạn chắc chắn rằng nó sẽ tạo ra một trận đấu). Nếu không tìm thấy kết quả phù hợp, dòng trên sẽ tạo ra một AttributeError nói rằng NoneType không thuộc tính group.

Answer 2

Vì vậy, bạn chưa để lại bất kỳ mô tả nào về những tệp này và cách bạn nhận chúng, nhưng tôi cho rằng bạn sẽ nhận được tên tệp bằng mô-đun os.

Như để nhận được số ra trong những cái tên, bạn muốn được tốt nhất bằng cách sử dụng biểu thức thông thường với re, một cái gì đó như thế này:

import re 
def get_numbers_from_filename(filename): 
    return re.search(r'\d+', filename).group(0) 

Sau đó, để bao gồm rằng trong một vòng lặp for, bạn muốn chạy chức năng đó trên mỗi tên tệp:

for filename in os.listdir(myfiledirectory): 
    print get_numbers_from_filename(filename) 

hoặc điều gì đó dọc theo các dòng đó.

Answer 3

Nếu có chỉ là một trong số:

filter(lambda x: x.isdigit(), filename)