有時我們會遇到將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]