时间戳

首页 > 解决方案 > 多标签数据的计数图

python - 多标签数据的计数图

问题描述

如何制作多标签数据的计数图？我有一个带有“流派”列的熊猫数据集；一部电影可能有不止一种类型。我需要一个带有不同类别的条的计数图（seaboorn.countplot 会做的事情）。

    Genre   
0   ['drama', 'comedy']
1   ['action']
2   ['drama']
3   ['action', 'comedy']

我尝试了 seaborn countplot，但它似乎不适用于标签列表。

标签： pythonpandasplotseabornvisualization

解决方案

如果数据是嵌套列表，则首先展平，转换为 DataFrame 并绘制：

import ast
#if necessary convert data to lists form strings
#df['Genre'] = df['Genre'].apply(ast.literal_eval)

df = pd.DataFrame({'a':[y for x in df['Genre'] for y in x]})
sns.countplot(x='a', data=df)

推荐阅读