[英]Extract indices of certain values from a list in python
假设,我有一个列表 [0.5,1,1.5,2,2.5,3,3.5,4,4.5],现在我想提取列表 [1.5,2.5,3.5,4.5] 的索引,这是一个该列表的子集。
您可以使用内置的 function <list>.index
在第一个列表中查找(第二个列表的)每个元素的索引。
了解了这一点后,您可以使用列表理解来实现您想要的:
>>> list1 = [0.5,1,1.5,2,2.5,3,3.5,4,4.5]
>>> list2 = [1.5,2.5,3.5,4.5]
>>> [list1.index(elem) for elem in list2]
[2, 4, 6, 8]
另一种选择是使用enumerate
function。请参阅以下答案:
a = [0.5,1,1.5,2,2.5,3,3.5,4,4.5]
b = [1.5,2.5,3.5,4.5]
indexes = []
for id, i in enumerate(a):
if i in b:
indexes.append(id)
print(indexes)
output 将是[2, 4, 6, 8]
。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.