python列表映射元素替换

有时我们会遇到将list中的元素替换为另一组元素的情况，此时可以采用列表映射进行快速元素替换。

例如： a = [2, 3, 4, 5, 0, 1] 需要将元素顺序替换为编号0， 1， 2， 3， 4， 5，此时就可以采用列表映射的方式。

# 通过列表映射进行元素替换
a = [2, 3, 4, 5, 0, 1]
map_ = {a[i]:i for i in range(len(a))}  # 映射字典
a_map = [map_[ele] if ele in map_ else ele for ele in a]