在python中提取两个不同字符串之间的int

Question

我有一个files字符串列表，格式如下：

files = ['/misc/lmbraid17/bensch/u-net-3d/2dcellnet/2dcellnet_v6w4l1/2dcellnet_v6w4l1_snapshot_iter_418000.caffemodel.h5', 
'/misc/lmbraid17/bensch/u-net-3d/2dcellnet/2dcellnet_v6w4l1/2dcellnet_v6w4l1_snapshot_iter_502000.caffemodel.h5', ...]

我想提取iter_和.caffemodel之间的int，并返回这些int的列表。你知道吗

经过一番研究，我想出了这个解决方案，但我想知道是否有一个更优雅的/Python式的方法，可能使用列表理解？你知道吗

li = []
for f in files:
   tmp = re.search('iter_[\d]+.caffemodel', f).group()
   li.append(int(re.search(r'\d+', tmp).group()))

Answer 1

也可以使用lookbehind assertion：

regex = re.compile("(?<=iter_)\d+")

for f in files:
     number = regex.search(f).group(0)