Merge pull request matplotlib#26937 from ksunden/scatter_c_type

Add ArrayLike to scatter c arg type hint

Author: QuLogic

lib/matplotlib/axes/_axes.pyi

@@ -402,7 +402,7 @@ class Axes(_AxesBase):
         x: float | ArrayLike,
         y: float | ArrayLike,
         s: float | ArrayLike | None = ...,
-        c: Sequence[ColorType] | ColorType | None = ...,
+        c: ArrayLike | Sequence[ColorType] | ColorType | None = ...,
         marker: MarkerType | None = ...,
         cmap: str | Colormap | None = ...,
         norm: str | Normalize | None = ...,

lib/matplotlib/pyplot.py

@@ -3670,7 +3670,7 @@ def scatter(
     x: float | ArrayLike,
     y: float | ArrayLike,
     s: float | ArrayLike | None = None,
-    c: Sequence[ColorType] | ColorType | None = None,
+    c: ArrayLike | Sequence[ColorType] | ColorType | None = None,
     marker: MarkerType | None = None,
     cmap: str | Colormap | None = None,
     norm: str | Normalize | None = None,