在seaborn heatmap中突出显示一行

import seaborn as sns
import matplotlib.pyplot as plt

# 导入数据集
data = sns.load_dataset("flights")

# 将数据集转换为透视表形式
pivot_data = data.pivot("month", "year", "passengers")

# 创建热力图
sns.heatmap(pivot_data, cmap="YlGnBu")

# 获取要突出显示的行索引
highlight_row = pivot_data.index.get_loc("June")

# 在热力图中突出显示一行
sns.heatmap(pivot_data, cmap="YlGnBu", mask=pivot_data.index != "June", cbar=False)

# 添加行标题
plt.xticks(ticks=range(len(pivot_data.columns)), labels=pivot_data.columns)
plt.yticks(ticks=range(len(pivot_data.index)), labels=pivot_data.index)

# 添加突出显示的行标记
plt.axhline(highlight_row, color='red', linewidth=3)