DOC: use notebook-style for plot_pca_3d.py (#23064)

nat-salt · lesteve · web-flow · commit fbc56edab96c · 2022-04-08T16:32:52.000+02:00
Co-authored-by: Loïc Estève &lt;loic.esteve@ymail.com&gt;
diff --git a/examples/decomposition/plot_pca_3d.py b/examples/decomposition/plot_pca_3d.py
@@ -15,15 +15,13 @@
 #          Kevin Hughes
 # License: BSD 3 clause
 
-from sklearn.decomposition import PCA
+# %%
+# Create the data
+# ---------------
 
 import numpy as np
-import matplotlib.pyplot as plt
-from scipy import stats
 
-
-# #############################################################################
-# Create the data
+from scipy import stats
 
 e = np.exp(1)
 np.random.seed(4)
@@ -51,8 +49,18 @@ def pdf(x):
 b /= norm
 
 
-# #############################################################################
+# %%
 # Plot the figures
+# ----------------
+
+from sklearn.decomposition import PCA
+
+import matplotlib.pyplot as plt
+
+# unused but required import for doing 3d projections with matplotlib < 3.2
+import mpl_toolkits.mplot3d  # noqa: F401
+
+
 def plot_figs(fig_num, elev, azim):
     fig = plt.figure(fig_num, figsize=(4, 3))
     plt.clf()
