繁体   English   中英

如果条件为真,则创建具有相邻列表元素的元组列表

[英]Create a list of tuples with adjacent list elements if a condition is true

我正在尝试创建一个元组列表,其中元组内容为数字9 ,列表中的数字为前面的数字。

输入清单:

myList = [1, 8, 9, 2, 4, 9, 6, 7, 9, 8]

期望的输出:

sets = [(8, 9), (4, 9), (7, 9)]

码:

sets = [list(zip(myList[i:i], myList[-1:])) for i in myList if i==9]

目前的结果:

[[], [], []]

更清洁的Pythonic方法:

>>> [(x,y) for x,y in zip(myList, myList[1:]) if y == 9]
[(8, 9), (4, 9), (7, 9)]

上面的代码是做什么的:

  • zip(some_list, some_list[1:])将生成一对相邻元素的列表。
  • 现在使用该元组,在第二个元素等于9的条件下进行过滤。 你完成了 :)

您的部分问题是myList[i:i]将始终返回一个空列表。 切片的结尾是独占的,所以当你执行a_list[0:0]你试图获取索引0和索引0 之间存在的a_list元素。

你是在正确的轨道上,但你想要自己压缩列表。

[(x, y) for x, y in zip(myList, myList[1:]) if y==9]

你非常接近,如果你刚刚开始,我会告诉你另一种可能更直观的方式:

sets = [(myList[i-1], myList[i]) for i in range(len(myList)) if myList[i] == 9]

获取列表长度范围内的索引,如果位置i值等于9 ,则抓取相邻元素。

结果是:

sets
[(8, 9), (4, 9), (7, 9)]

比其他方法效率低,但我决定取消删除它以向您展示不同的方法。 你可以使用enumerate()代替它:

sets = [(myList[i-1], j) for i, j in enumerate(myList) if j == 9]

请注意myList[0] = 9的边缘情况下,没有zip的理解行为和zip的理解行为是不同的

具体来说,如果myList = [9, 1, 8, 9, 2, 4, 9, 6, 7, 9, 8]那么:

[(myList[i-1], myList[i]) for i in range(len(myList)) if myList[i] == 9]
# results in: [(8, 9), (8, 9), (4, 9), (7, 9)]

而:

[(x, y) for x, y in zip(myList, myList[1:]) if y==9]
# results in: [(8, 9), (4, 9), (7, 9)]

由你来决定哪一个符合你的标准,我只是指出它们在所有情况下的表现都不一样。

您也可以通过创建迭代器来完成它而不进行切片:

l = myList = [1,8,9,2,4,9,6,7,9,8]

it1, it2 = iter(l), iter(l)
# consume first element from it2 -> leaving 8,9,2,4,9,6,7,9,8
next(it2, "")
# then pair up, (1,8), (8,9) ...
print([(i, j) for i,j in zip(it1, it2) if j == 9])

或者使用成对配方来创建配对

from itertools import tee, izip

def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return izip(a, b)

如果使用python3,只需导入tee并使用常规zip

令人惊讶的是,没有人添加功能方法。

另一个替代答案是使用filter 这个内置函数返回一个迭代器(Python2中的列表),该迭代器包含列表中存在的所有元素,这些元素为特定函数返回True

>>> myList = [1,8,9,2,4,9,6,7,9,8]
>>> list(filter(lambda x:x[1]==9,zip(myList, myList[1:])))
[(8, 9), (4, 9), (7, 9)]

需要注意的是,只有在python3 +中才需要list调用 本文详细讨论了功能方法和列表推导之间的区别。

我的解决方案类似于Jim的高级零索引检查

myList = [9, 1, 8, 9, 2, 4, 9, 6, 7, 9, 8]

[(myList[i-1], x) for i, x in enumerate(myList) if x==9 and i!=0]

# [(8, 9), (4, 9), (7, 9)]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM