将字符串列表映射到熊猫列?

时间:2021-05-02 16:54:07

标签: python pandas

只是古玩更好的方法来映射熊猫列与列表。

ref_list=['a','b','c','d']
lst = [0,2,1]
df = pd.DataFrame(lst,columns=['no'])

预期输出

   no map
0   0  a
1   2  c
2   1  b

2 个答案:

答案 0 :(得分:3)

使用枚举字典映射:

df['map_'] = df['no'].map(dict(enumerate(ref_list)))
#df['map_'] = np.array(ref_list)[lst]

print(df)

   no map_
0   0    a
1   2    c
2   1    b

答案 1 :(得分:1)

df = pd.DataFrame(zip(lst, np.array(ref_list)[lst]), columns=["no", "map"])
print(df)

打印:

   no map
0   0   a
1   2   c
2   1   b