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]
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章