Создать тепловую карту с категориальными данными в python (Seaborn) - PullRequest
1 голос
/ 28 марта 2020

Я пытаюсь создать тепловую карту с использованием Seaborn, которая показывает экземпляры комбинаций типов в DataFrame, содержащем все данные Pokemon.

poke_df.head()

    NationalNo  Name        Type            Type_1  Type_2  HP  Total
0   001         Bulbasaur   [Grass, Poison] Grass   Poison  45  318 
1   002         Ivysaur     [Grass, Poison] Grass   Poison  60  405
2   003         Venusaur    [Grass, Poison] Grass   Poison  80  525

Вот что у меня есть, хотя вывод выглядит так: не близко к тому, что он должен.

from requests import get
from bs4 import BeautifulSoup
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import sys
import re
reload(sys)
sys.setdefaultencoding('utf-8')
%matplotlib inline

sns.set()


plt.figure(figsize=(16,8))
#flights = poke_df
flights = pd.pivot_table(poke_df, index="Type_1", columns="Type_2",fill_value=0)
sns.heatmap(flights)

Включенная картинка (сделанная в виде таблицы) более или менее подходит для меня.

Desired result

...