For 2d scatter, I can pass 2d arrays as X and Y. But it seems not c=.
Here is a demonstration:
···
import matplotlib.pyplot as plt
import numpy as np
A = np.array
import colorcet as cc
cmap = cc.m_rainbow
x = A([[0,.25], [0.5, 0.75]])
y = x**2
colors = np.empty (x.shape, dtype=object)
for i in range(2):
for j in range(2):
colors[i,j] = cmap(x[i,j])
plt.scatter (x, y, c=colors.flatten())
Without colors.flatten(), mpl will give an error. It seems to be OK with X and Y being 2D, but not c.
In this example I have given c as a 2d array of objects, each object being an RGBA tuple.
Is there an easier way? Preferably one that doesn’t require me to specifically allocate an empty array of objects
and then populate it with an explicit loop?