根据第二个列表获取列表索引

时间:2019-07-04 16:16:09

标签: python

我有一个列表a,其中包含列表b中的所有可能值

a = ['foo', 'bar', 'baz']

b = ['baz', 'baz', 'foo', 'foo', 'foo', 'bar', 'foo', 'baz']

我想返回一个列表c,该列表具有在b中找到的元素数量,其中每个元素都是a的索引,其中{{1 }}。

示例

b

2 个答案:

答案 0 :(得分:1)

单线:

c = [a.index(x) for x in b]

答案 1 :(得分:1)

>>> a = ['foo', 'bar', 'baz']
>>> b = ['baz', 'baz', 'foo', 'foo', 'foo', 'bar', 'foo', 'baz']
>>> [a.index(i) for i in b]
[2, 2, 0, 0, 0, 1, 0, 2]