热图中的下标标签（在Python seaborn中）

Question

我正在尝试在 Python/sns 热图中为我的轴标签添加下标；但是，我不知道该怎么做。下面是我获取热图的示例代码，但标签是正常的（附图），但我希望它们作为下标。例如，我想将 Tair 作为下标中的空气。另一个例子是 Tair-Ts 变量，我想要空气和 s 位于下标中。

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

d = {'Tair': [25.55,  5.25, 1.99,  9.78], 
     'Ts': [0.45,  0.10,  0.69, 0.52],
    'Tair-Ts': [-10, -7, -8, -5]}

df = pd.DataFrame(data=d) # data frame
corr = df.corr() # correlation matrix
ax = sns.heatmap(corr, linewidth=0.5) # heatmap

如果有人能帮助我将标签作为下标，我将非常感激。

谢谢你。

示例图

Answer 1

在 matplotlib 中，您可以使用美元符号来指示 LaTeX 格式。

您的示例可能如下所示：

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

d = {'Tair': [25.55, 5.25, 1.99, 9.78],
     'Ts': [0.45, 0.10, 0.69, 0.52],
     'Tair-Ts': [-10, -7, -8, -5]}

df = pd.DataFrame(data=d)
corr = df.corr()
labels = ['$T_{air}$', '$T_{s}$', '$T_{air}-T_{s}$']
ax = sns.heatmap(corr, xticklabels=labels, yticklabels=labels, linewidth=0.5)
plt.tight_layout()
plt.show()