短小精悍算例:Python如何提取列表list中指定元素的位置索引

列表a=[1, 2, 3, 2, 2, 3, 0, 8, 3]
现要找出列表a中元素3所在的索引位置。
可以使用a.index(3),但是只能返回第一个元素3的索引,不能返回所有元素3的索引。

a=[1, 2, 3, 2, 2, 3, 0, 8, 3]
a.index(3)
返回结果:2     # 可见只返回第一个元素3的索引。

考虑用enumerate方法,如下:

a=[1, 2, 3, 2, 2, 3, 0, 8, 3]
r1 = [k for (k,v) in enumerate(a) if v == 3]
print(r1)
返回结果:[2, 5, 8]

或者使用range和len()方法如下:

a=[1, 2, 3, 2, 2, 3, 0, 8, 3]
r2 = [i for i in range(len(a)) if a[i] == 3]
print(r2)
返回结果:[2, 5, 8]

可见,Python能巧妙使用list和for循环,进而解决很多问题。

猜你喜欢

转载自blog.csdn.net/weixin_39464400/article/details/106067096