如何用Python编写程序来生成热力图？

本文共计793个文字，预计阅读时间需要4分钟。

1. 引言+热力图，是一种通过颜色块来展示数据统计的图表。绘图时，需指定颜色映射规则。例如，较大的值用较深的颜色表示，较小的值用较浅的颜色表示；较大的值用偏暖色调表示。

1. 引言

热力图，是一种通过对色块着色来显示数据的统计图表。绘图时，需指定颜色映射的规则。例如，较大的值由较深的颜色表示，较小的值由较浅的颜色表示；较大的值由偏暖的颜色表示，较小的值由较冷的颜色表示，等等。

本文主要用Python来实现热力图的制作，样例如下：

闲话少说，我们直接开始吧！

2. 读取图像

首先，我们来读取样例图像，并对其进行相应的​​crop​​操作。样例代码如下：

import numpy as np
import matplotlib.pyplot as plt
import skimage.io as io
img = plt.imread("butterfly-7320158_960_720.jpg")
# crop
img_cut = img[147:447,:,:]
plt.figure()
plt.imshow(img_cut)

运行后，得到结果如下：

3. 图像灰度化

由于热力图主要用于可视化二维矩阵，所以我们需要将彩色图转化为灰度图，代码如下：

from skimage.color import rgb2gray
gray_img = rgb2gray(img_cut)
plt.imshow(gray_img,cmap='gray')

运行结果如下：

4. 像素化效果

这一步的操作主要是为了使图像看起来像像素化的艺术品。