палитра matplotlib для разброса

Я работаю с данными, у которых есть 3 параметра построения графика: x, y, c. Как создать собственное значение цвета для диаграммы рассеяния?

Расширение этого примера Я пытаюсь сделать:

import matplotlib
import matplotlib.pyplot as plt
cm = matplotlib.cm.get_cmap('RdYlBu')
colors=[cm(1.*i/20) for i in range(20)]
xy = range(20)
plt.subplot(111)
colorlist=[colors[x/2] for x in xy] #actually some other non-linear relationship
plt.scatter(xy, xy, c=colorlist, s=35, vmin=0, vmax=20)
plt.colorbar()
plt.show()

, но результат TypeError: сначала необходимо set_array для отображаемого