matplotlib.pyplot画热力图

以下是两种使用matplotlib.pyplot绘制热力图的方法：

1. 使用matplotlib.pyplot.matshow()

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(10, 10)
plt.matshow(x, cmap=plt.cm.winter, vmin=0, vmax=1)
plt.colorbar()
plt.show()

2. 使用matplotlib.pyplot.imshow()

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(10, 10)
plt.imshow(x, cmap=plt.cm.winter, vmin=0, vmax=1)
plt.colorbar()
plt.show()

相关问题

matplotlib.pyplot 绘画热力图时调整数字字体的大小

要调整热力图中数字的字体大小，可以使用 `text` 函数。`text` 函数可以在指定的坐标上添加一个文本标签，包括数字和字体样式等信息。

例如，下面的代码将热力图中数字的字体大小设置为 10：

import numpy as np
import matplotlib.pyplot as plt

# 生成数据
data = np.random.rand(5, 5)

# 绘制热力图
fig, ax = plt.subplots()
im = ax.imshow(data, cmap='coolwarm')

# 添加数字标签
for i in range(data.shape[0]):
    for j in range(data.shape[1]):
        text = ax.text(j, i, round(data[i, j], 2),
                       ha="center", va="center", color="w", fontsize=10)

ax.set_title("Heatmap with Annotated Numbers")

# 调整热力图的布局
fig.tight_layout()

plt.show()

在上面的代码中，我们首先使用 `imshow` 函数绘制热力图，然后使用两个嵌套的 `for` 循环为热力图添加数字标签。接着，我们调用 `text` 函数，并将字体大小设置为 10。最后，使用 `tight_layout` 函数调整热力图的布局。运行上面的代码，可以得到一个带有数字标签的热力图，其中数字的字体大小为 10。

matplotlib.pyplot 绘画热力图时调整表格中数字字体的大小

要调整热力图中表格中数字的字体大小，可以使用 `annotate` 函数。`annotate` 函数可以在指定的坐标上添加一个文本标签，包括数字和字体样式等信息。

例如，下面的代码将热力图中表格中数字的字体大小设置为 10：

import numpy as np
import matplotlib.pyplot as plt

# 生成数据
data = np.random.rand(5, 5)

# 绘制热力图
fig, ax = plt.subplots()
im = ax.imshow(data, cmap='coolwarm')

# 添加数字标签
for i in range(data.shape[0]):
    for j in range(data.shape[1]):
        text = ax.text(j, i, round(data[i, j], 2),
                       ha="center", va="center", color="w")

ax.set_title("Heatmap with Annotated Numbers")

# 调整数字标签的字体大小
for text in ax.texts:
    text.set_fontsize(10)

# 调整热力图的布局
fig.tight_layout()

plt.show()

在上面的代码中，我们首先使用 `imshow` 函数绘制热力图，然后使用两个嵌套的 `for` 循环为热力图添加数字标签。接着，我们使用一个 `for` 循环遍历所有的数字标签，并调用 `set_fontsize` 方法将数字标签的字体大小设置为 10。最后，使用 `tight_layout` 函数调整热力图的布局。运行上面的代码，可以得到一个带有数字标签的热力图，其中表格中数字的字体大小为 10。