From b7c34515b9b65978956f8eb9716f6fe0984626c9 Mon Sep 17 00:00:00 2001
From: Tim Hoffmann <2836374+timhoffm@users.noreply.github.com>
Date: Fri, 2 Aug 2024 00:34:57 +0200
Subject: [PATCH] DOC: Simplify heatmap example

Directly rotate ticks in `set_xticks` instead of setting the ticks and
then changing them afterwards using `plt.setp(...)`.
---
 .../image_annotated_heatmap.py                 | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/galleries/examples/images_contours_and_fields/image_annotated_heatmap.py b/galleries/examples/images_contours_and_fields/image_annotated_heatmap.py
index 23d9fd48dff8..ebd50359a29a 100644
--- a/galleries/examples/images_contours_and_fields/image_annotated_heatmap.py
+++ b/galleries/examples/images_contours_and_fields/image_annotated_heatmap.py
@@ -63,12 +63,9 @@
 im = ax.imshow(harvest)
 
 # Show all ticks and label them with the respective list entries
-ax.set_xticks(np.arange(len(farmers)), labels=farmers)
-ax.set_yticks(np.arange(len(vegetables)), labels=vegetables)
-
-# Rotate the tick labels and set their alignment.
-plt.setp(ax.get_xticklabels(), rotation=45, ha="right",
-         rotation_mode="anchor")
+ax.set_xticks(range(len(farmers)), labels=farmers,
+              rotation=45, ha="right", rotation_mode="anchor")
+ax.set_yticks(range(len(vegetables)), labels=vegetables)
 
 # Loop over data dimensions and create text annotations.
 for i in range(len(vegetables)):
@@ -137,17 +134,14 @@ def heatmap(data, row_labels, col_labels, ax=None,
     cbar.ax.set_ylabel(cbarlabel, rotation=-90, va="bottom")
 
     # Show all ticks and label them with the respective list entries.
-    ax.set_xticks(np.arange(data.shape[1]), labels=col_labels)
-    ax.set_yticks(np.arange(data.shape[0]), labels=row_labels)
+    ax.set_xticks(range(data.shape[1]), labels=col_labels,
+                  rotation=-30, ha="right", rotation_mode="anchor")
+    ax.set_yticks(range(data.shape[0]), labels=row_labels)
 
     # Let the horizontal axes labeling appear on top.
     ax.tick_params(top=True, bottom=False,
                    labeltop=True, labelbottom=False)
 
-    # Rotate the tick labels and set their alignment.
-    plt.setp(ax.get_xticklabels(), rotation=-30, ha="right",
-             rotation_mode="anchor")
-
     # Turn spines off and create white grid.
     ax.spines[:].set_visible(False)