get array of colors instead of colormap object

neuropsychology · May 20, 2024 · 836f62a · 836f62a
1 parent b3e3828
commit 836f62a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/neurokit2/complexity/entropy_phase.py b/neurokit2/complexity/entropy_phase.py
@@ -123,7 +123,8 @@ def entropy_phase(signal, delay=1, k=4, show=False, **kwargs):
         Tx = Tx.astype(bool)
         Ys = np.sin(angles) * limx * np.sqrt(2)
         Xs = np.cos(angles) * limx * np.sqrt(2)
-        colors = plt.get_cmap("jet").resampled(k)
+        resampled_cmap = plt.get_cmap("jet").resampled(k)
+        colors = resampled_cmap(np.linspace(0, 1, k))
 
         plt.figure()
         for i in range(k):