gpt4 book ai didi

python - 多标签数据的计数图

转载 作者:行者123 更新时间:2023-12-04 10:53:03 25 4
gpt4 key购买 nike

如何制作多标签数据的计数图?我有一个带有“流派”列的 Pandas 数据集;一部电影可能有不止一种类型。我需要一个带有不同类别条形图的计数图(seaboorn.countplot 会做的事情)。

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

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

最佳答案

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

关于python - 多标签数据的计数图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59372541/

25 4 0