Python:将 list.index 与正则表达式一起使用

Question

我有一些字符串列表，我想提取其中的某个值:

["bla","blabla","blablabla","time taken to build model: 5.1 seconds", "blabla"]

通常我会通过

查找我要查找的元素的索引

list.index("time taken")

但是由于时代的变化，我想到了使用正则表达式。我只是不知道该怎么做。

那么我怎样才能找到与某个正则表达式匹配的列表元素的索引，例如重新匹配()？(如果不遍历列表，这将花费很长时间)

Answer 1

不确定是否有内置方法，但使用列表推导式可以在 O(n) 时间内轻松完成此操作。

使用正则表达式:

import re
your_list = ["bla","blabla","blablabla","time taken to build model: 5.1 seconds", "blabla"]
regex = re.compile("^time taken")
idxs = [i for i, item in enumerate(your_list) if re.search(regex, item)]

没有正则表达式:

your_list = ["bla","blabla","blablabla","time taken to build model: 5.1 seconds", "blabla"]
query_term = 'time taken'
idxs = [i for i, item in enumerate(your_list) if item.startswith(query_term)]

您可以让它返回第一个找到的索引或最后找到的索引，或者在方法中对其进行参数化以提供灵 active 。