python - 过滤列表的优雅方式

Question

我有两个类似以下方式的列表。我需要从 = 即 xyz 和 .81 之后提取值。我正在尝试为列表或未来制定一个通用解决方案。截至目前，我正在加入并找到列表而不是拆分。但我觉得它可以以更好的方式完成。

a = ['.TITLE', "'=", 'xyz', 'vdd', '=', '0.81', 'temp', "-40'"]
b = ['.TITLE', "'$", 'abc', '=', 'xyz', 'vdd', '=', '0.99', 'temp', "125'"]

Answer 1

您可以使用next()生成器：

a = ['.TITLE', "'=", 'xyz', 'vdd', '=', '0.81', 'temp', "-40'"]
a1 = iter(a)
for item in a1:
    if '=' in item:
        print next(a1)

印刷：

xyz
0.81

请注意，如果=是最后一项，这将引发StopIteration错误。

---

你也可以使用enumerate()：

a = ['.TITLE', "'=", 'xyz', 'vdd', '=', '0.81', 'temp', "-40'"]
for i, j in enumerate(a):
    if '=' in j:
        print a[i+1]

印刷：

xyz
0.81

IndexError如果=是最后一项，这将再次引发错误 ( )。

Answer 2

作为一个函数：

import itertools

def extract_items_after(lst, marker='='):
    for x, y in itertools.izip_longest(lst, lst[1:]):
        if marker in x:
            yield y

使用您的数据：

a = ['.TITLE', "'=", 'xyz', 'vdd', '=', '0.81', 'temp', "-40'"]
b = ['.TITLE', "'$", 'abc', '=', 'xyz', 'vdd', '=', '0.99', 'temp', "125'"]

for l in [a, b]:
    print list(extract_items_after(l))

结果：

>>> 
['xyz', '0.81']
['xyz', '0.99']

Answer 3

一种非优雅的方式是使用标志：

a = ['.TITLE', "=", 'xyz', 'vdd', '=', '0.81', 'temp', "-40'"]
b = ['.TITLE', "'$", 'abc', '=', 'xyz', 'vdd', '=', '0.99', 'temp', "125'"]

def extract(L,sep="="):
    pick_this_one = False
    for item in L :
        if pick_this_one:
            pick_this_one = False
            yield item
        if item == sep :
            pick_this_one = True

for value in extract(a+b):
    print value

>>> xyz
0.81
xyz
0.99
>>>