Я пытаюсь показать цветную шкалу моего графика рассеяния, но я получаю ошибку:

TypeError: сначала вы должны установить set_array для mappable

Вот что я делаю для сюжета:

# Just plotting the values of data that are nonzero 
x_data = numpy.nonzero(data)[0] # x coordinates
y_data = numpy.nonzero(data)[1] # y coordinates

# Mapping the values to RGBA colors
data = plt.cm.jet(data[x_data, y_data])

pts = plt.scatter(x_data, y_data, marker='s', color=data)

plt.colorbar(pts)

Если я прокомментирую строкуplt.colorbar(pts) Я получил график правильно, но я бы тоже хотел построить цветную полосу.

Заранее спасибо.