多标签数据的计数图

时间:2019-12-17 10:46:08

标签: python pandas plot seaborn visualization

如何制作多标签数据的计数图?我有一个列为“ Genre”的熊猫数据集;电影的类型可能不止一种。我需要一个带有用于不同类的条的计数图(seaboorn.countplot可以执行的操作)。

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

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

1 个答案:

答案 0 :(得分:2)

如果数据是嵌套列表,则首先展平,转换为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)