pandas: Как построить круговую диаграмму для количества фильмов в зависимости от жанра фильмов IMDB в pandas?

Question

У меня есть следующий набор данных:

import pandas as pd
import numpy as np 
%matplotlib inline

df = pd.DataFrame({'movie' : ['A', 'B','C','D'], 
                   'genres': ['Science Fiction|Romance|Family', 'Action|Romance',
                              'Family|Drama','Mystery|Science Fiction|Drama']},
                  index=range(4))
df

Моя попытка

# Parse unique genre from all the movies
gen = []
for g in df['genres']:
    gg = g.split('|')
    gen = gen + gg
    gen = list(set(gen))

print(gen)

df['genres'].value_counts().plot(kind='pie')

Я получил это изображение:

Но я бы хотел сделать круговую диаграмму для каждого отдельного жанра.

Как мы получаем жанры для подсчета количества фильмов для каждого уникального жанра?

Answer 1

Вы можете сделать .str.split() с expand=True, что даст вам DataFrame всех жанров. Если вы затем сложите это, вы получите значения для всех жанров.

df.genres.str.split('|', expand=True).stack().value_counts().plot(kind='pie', label='Genre')

Это может быть немного медленнее для вычисления количества, поэтому более быстрая реализация для того же графика будет (с добавлением процентов):

from itertools import chain
from collections import Counter
import matplotlib.pyplot as plt

cts = Counter(chain.from_iterable(df.genres.str.split('|').values))
_ = plt.pie(cts.values(), labels=cts.keys(), autopct='%1.0f%%')
_ = plt.ylabel('Genres')

Answer 2

Итак, однострочное решение:

df.genres.str.get_dummies().sum().plot.pie(label='Genre', autopct='%1.0f%%')

Результат:

---

TL; DR

Во-первых, преобразуйте столбец категорий в макеты:

df = pd.concat([df.drop('genres', axis=1), df.genres.str.get_dummies()], axis=1)

Результат:

  movie  a  b  c  d  e  f  g
0     A  1  1  1  0  0  0  0
1     B  0  0  1  0  1  0  0
2     C  0  0  0  0  0  1  1
3     D  1  1  0  1  1  0  0

Затем подсчитайте количество вхождений для каждой категории:

counts = df.drop('movie', axis=1).sum()

Результат:

a    2
b    2
c    2
d    1
e    2
f    1
g    1

И, наконец, построите круговую диаграмму:

counts.plot.pie()