我正在尝试使用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制作)或多或少是我想要的。