使用python中的分类数据创建热图(Seaborn)

时间:2020-03-28 06:07:20

标签: python seaborn heatmap

我正在尝试使用Seaborn创建一个热图,以显示包含所有Pokemon数据的DataFrame中类型组合的实例。

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)

包含的图片(用Tableau制作)或多或少是我想要的。

Desired result

0 个答案:

没有答案