如何在Seaborn热力图中仅显示满足条件的X轴刻度标签

核心思路 #

先筛选出所有行值均≥2的列，然后给热力图设置自定义X轴标签——符合条件的列保留原标签，不符合的设为空（或直接隐藏对应刻度），同时完整保留热力图的所有单元格绘制。

完整实现代码 #

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

# 构造示例数据
df = pd.DataFrame({'set1': ['0', '2', '2'],
                   'set2': ['1', '2', '0'],
                   'set3': ['0', '2', '1'],
                   'set4': ['1', '4', '1']
                  }).T.astype(float)

# 1. 筛选符合条件的列：所有行值≥2的列
valid_columns = df.ge(2).all(axis=0)

# 2. 生成自定义X轴标签：符合条件的留原标签，否则设为空字符串
custom_xtick_labels = [col if is_valid else '' for col, is_valid in zip(df.columns, valid_columns)]

# 3. 绘制热力图
plt.figure(figsize=(8, 4))
ax = sns.heatmap(df, yticklabels=df.index, xticklabels=custom_xtick_labels)

# 可选优化：彻底隐藏无标签的刻度线（避免空标签位置留刻度线）
# 获取当前X轴的刻度位置和标签
current_ticks = ax.get_xticks()
current_labels = ax.get_xticklabels()
# 只保留有有效标签的刻度
filtered_ticks = [tick for tick, label in zip(current_ticks, current_labels) if label.get_text() != '']
filtered_labels = [label for label in current_labels if label.get_text() != '']
# 更新X轴刻度和标签
ax.set_xticks(filtered_ticks)
ax.set_xticklabels(filtered_labels)

plt.tight_layout()
plt.show()

代码细节解释 #

• df.ge(2).all(axis=0)：逐列检查每一列的所有元素是否都≥2，返回一个布尔值Series，标记每列是否符合要求。
• custom_xtick_labels：通过列表推导式生成标签列表，把不符合条件的列标签替换为空字符串，这样绘制时这些位置就不会显示标签。
• 可选的刻度线隐藏：如果不想保留空标签对应的刻度线，通过重新设置X轴的刻度位置和标签，只保留有有效标签的部分，让X轴更整洁。

这样处理后，热力图会完整展示所有单元格，仅隐藏不符合条件的X轴刻度标签（或连带刻度线），有效解决X轴拥挤的问题。

内容的提问来源于stack exchange，提问作者Caterina