逆标签编码用于绘图

时间:2019-05-03 17:39:38

标签: python pandas

我在DataFrame中具有以整数编码的功能,即int64类型。此外,我有一本包含等效类的字典。有没有办法用原始类别为此功能绘制条形图?

df = pd.DataFrame({“ feature_1”:[-1,0,1,1,0,1]})

标签= {“好”:1,“坏”:0,“独立”:-1}

我想获得一种饼形图,其中包含每种情况的相对频率(%),并且字典中包含字符串标签。

1 个答案:

答案 0 :(得分:0)

这应该为您工作。只需操纵df即可替换所需单词的值,然后进行计数。

import matplotlib.pyplot as plt
import seaborn as sns

df = pd.DataFrame({"feature_1": [-1, 0, 1, 1, 0, 1]})
tags = {"Good": 1, "Bad": 0, "Indiferent":-1}


df = df['feature_1'].replace({v:k for k,v in tags.items()})\
                    .value_counts()\
                    .reset_index()\
                    .rename(columns = {'index': 'feature_1','feature_1':'count'})

sns.catplot(x = "feature_1", y = 'count', data = df, kind = 'bar')

enter image description here