将颜色条添加到 2d hist - Python

在Python中，我们经常需要绘制二维直方图。但是，如何添加颜色条到二维直方图里呢？实际上，这个问题并不复杂。下面我们来介绍如何用Python在二维直方图中添加颜色条。

1. 导入matplotlib和numpy库

首先，我们需要导入绘图库matplotlib和数学库numpy。具体示例代码如下：

import matplotlib.pyplot as plt
import numpy as np

2. 准备数据

为了演示，我们需要先准备一些数据。下面是一个用numpy生成的二维正态分布的数据。

# generate 2D normal distribution data
mean = [0, 0]
cov = [[1, 0], [0, 1]]
x, y = np.random.multivariate_normal(mean, cov, 10000).T

3. 绘制二维直方图

接下来，我们可以使用matplotlib中的hist2d()函数来绘制二维直方图。示例代码如下：

plt.hist2d(x, y, bins=100)
plt.colorbar()
plt.show()

在上述代码中，hist2d()函数用来绘制二维直方图，其中参数x和y是数据，bins是指定分成多少个小的正方形格子，这里设置了100。另外，colorbar()函数可以用来添加颜色条。

4. 完整代码

最终，我们得到了完整的代码。下面是示例代码的汇总：

import matplotlib.pyplot as plt
import numpy as np

# generate 2D normal distribution data
mean = [0, 0]
cov = [[1, 0], [0, 1]]
x, y = np.random.multivariate_normal(mean, cov, 10000).T

# plot 2D histogram
plt.hist2d(x, y, bins=100)
plt.colorbar()
plt.show()

我们可以运行上述代码，得到一个带有颜色条的二维直方图。