pandas numpy 映射问题

最新推荐文章于 2024-04-25 17:02:00 发布

我的心永远是冰冰哒

最新推荐文章于 2024-04-25 17:02:00 发布

阅读量257

点赞数

分类专栏： python 文章标签： python 开发语言后端

本文链接：https://blog.csdn.net/qq_45759229/article/details/121049298

版权

python 专栏收录该内容

101 篇文章 3 订阅

订阅专栏

dataframe 与dict 相互转化

import numpy as np
import pandas as pd 
## 创建一个dataframe表示映射关系，我一般都是从csv文件中读取
map_df=pd.DataFrame({"int":[1,2,3],"celltype":["spk","vis","wal"]})
print(map_df)
print("*"*50)
## 从dataframe重新生成对应的关系
celltype_map_dict=map_df.set_index('int')['celltype'].to_dict()
print(celltype_map_dict)

在这里插入图片描述

映射np.array

arr=np.array([1,2,2,3,1,2,2],dtype=int)
map_arr=np.vectorize(celltype_map_dict.get)(arr)
print(map_arr)
print(type(map_arr))

在这里插入图片描述

映射dataframe

pd_df=pd.DataFrame({"label":np.array([1,2,3,2,3,2,1,3,2,1,1,1,2,3])})
print(pd_df)
######################方法1#############################
pd_df["celltype1"]=pd_df["label"].map(celltype_map_dict)
print(pd_df)
######################方法2##############################
pd_df["celltype2"]=pd_df['label'].replace(celltype_map_dict, inplace=False)
print(pd_df)