python中找到所有指定元素的索引

定义:X=[1,2,3,1,4]
任务:找出元素为1的索引
Solution:

#如果直接用X.index(1),只能得到0这一个索引,而我们需要所有索引.
l = len(X)
zip_list = zip(*(range(l),X))
id1 = [z[0] for i,z in enumerate(zip_list) if z[1]==1]

# 或者更简单的
id1 = [i for i,x in enumerate(X) if x==1]

作者:Kerrwy
来源:CSDN
原文:https://blog.csdn.net/Kerrwy/article/details/82419132
版权声明:本文为博主原创文章,转载请附上博文链接!

猜你喜欢

转载自blog.csdn.net/qq_29007291/article/details/83511125