Skip to main content
 首页 » 编程设计

python之根据另一个列表元素获取一些列表元素索引

2025年05月04日14think

我有两个列表:

a=['book','car','car','have'] 
b=['car','have'] 

我想要这个输出:

a_basedon_b_indexes=[1,2,3]# 'car' and 'have' indexes in list a 

我想要这个输出的单行表达式。(我知道如何用 for 循环来做)。这在 python 中可能吗?

请您参考如下方法:

你可以在列表理解中使用enumerate

>>> [i for i,j in enumerate(a) if j in b] 
[1, 2, 3] 

如果 b 很大,我建议使用 set 因为 in 操作会更快

>>> b = {'car', 'have'} 
>>> [i for i,j in enumerate(a) if j in b] 
[1, 2, 3]