pandas 带计算字段的Python热图

mbzjlibv  于 2022-12-10  发布在  Python
关注(0)|答案(1)|浏览(146)

希望从 Dataframe 创建热图。索引是每个车祸事件。列包括年、月(1 - 12)、星期几(1- 7)、小时(0 - 23)、致命(1)非致命(2)等。
我试图创建一个热图,x轴是一天中的小时,y轴是一周中的天。希望为每个“单元格”创建一个计算字段,对应于每个小时和每天的死亡率。
第一个
我想我已经整理好了以获得我需要的数字,但是如何将它们分配给我正在寻找的热图呢?

7kjnsjlb

7kjnsjlb1#

首先,使用您的数据制作一个二维矩阵,其中的行表示日期(星期日、...),列表示数字(0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18)。
获得此2-D矩阵后,使用以下代码绘制热图

import numpy as np
import matplotlib.pyplot as plt

# create a 10x10 random matrix
data = np.random.random((10, 10))   # REPLACE WITH YOUR DATA
print(data.shape)

fig, ax = plt.subplots()
im = ax.imshow(data)

# show image
plt.show()

相关问题