从 Matplotlib 中的颜色栏中删除标签和刻度线

Question

我有一个热图，我想删除颜色条上的标签和刻度线。如何使用迄今为止制作的热图（下面的代码）来做到这一点？

import seaborn as sns
import matplotlib.pyplot as plt

sns.heatmap(my_heatmap, cmap='RdYlGn', vmin=0, vmax=1)
plt.rcParams["figure.figsize"] = (8, 6)
plt.title('10 mm Precipitation Probabilities\nTopeka, Kansas', fontsize = 16)
plt.xlabel('X', fontsize = 14)
plt.ylabel('Y', fontsize = 14)
plt.text(6.7,.18, 'High Probability')
plt.show()

（我宁愿它只是在栏顶部显示高概率，如我制作的标签所示）。

Answer 1

cbar = ax.collections[0].colorbar您可以通过（参见例如这篇文章）访问颜色栏。

然后，您可以更改刻度 ( cbar.set_ticks([])) 或设置新刻度和新刻度标签：

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = (8, 6)
my_heatmap = np.random.rand(10, 10)
ax = sns.heatmap(my_heatmap, cmap='RdYlGn', vmin=0, vmax=1)
# ax.get_figure().set_size_inches(8, 6)
ax.set_title('10 mm Precipitation Probabilities\nTopeka, Kansas', fontsize=16)
ax.set_xlabel('X', fontsize=14)
ax.set_ylabel('Y', fontsize=14)
cbar = ax.collections[0].colorbar
cbar.set_ticks([0, 0.5, 1])
cbar.set_ticklabels(['Low\nProbability', 'Average\nProbability', 'High\nProbability'])
# cbar.set_label('Probability')
plt.tight_layout()
plt.show()

请注意，这plt.rcParams["figure.figsize"]不会更改图形的大小，而是为将创建的下一个图形设置默认大小。sns.heatmap因此，如果之前没有创建过，您需要在创建一个新图形之前调用它。要更改已创建图形的大小，请使用set_size_inches()。