[英]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.