我如何从两列加上 python 中这两列的值创建热图

Question

感谢您的帮助！

我想根据数据 df 在 python 中生成热图。
（我在我的项目中使用 pandas、seaborn、numpy 和 matplotlib）

数据框 df 看起来像：

index | a | b | c | year | month
0     |   |   |   | 2013 |   1
1     |   |   |   | 2015 |   4
2     |   |   |   | 2016 |   10 
3     |   |   |   | 2017 |   1

在数据集中，每一行都是一张票。

数据集很大（51 列和 100k+ 行），
所以 a、b、c 只是为了显示一些随机列。（月份 => 1 = 1 月，2 = 2 月...）

对于热图：

x-axis = year,
y-axis = month,

值：在热图中，我希望两个轴之间的值是行数，其中一张票在当年和那个月已经给出。

我想象的结果应该类似于 seaborn 文档中的结果： https://seaborn.pydata.org/_images/seaborn-heatmap-4.png

我是编码新手，尝试了很多我在 Internet 上找到的随机方法，但一直无法正常工作。

感谢您的帮助！

Answer 1

应该这样做（使用生成的数据）：

import pandas as pd
import seaborn as sns
import random

y = [random.randint(2013,2017) for n in range(2000)]
m = [random.randint(1,12) for n in range(2000)]


df = pd.DataFrame([y,m]).T
df.columns=['y','m']
df['count'] = 1
df2 = df.groupby(['y','m'], as_index=False).count()
df_p = pd.pivot_table(df2,'count','m','y')


sns.heatmap(df_p)

您可能不需要 count 列，但我添加它是因为 groupby 需要一个额外的列。

我如何从两列加上 python 中这两列的值创建热图

How do i create a heatmap from two columns plus the value of those two in python

python

heatmap