[Python] Valores de Barras de Cores no Matplotlib

Eu tenho um código com 3 variáveis: a1, a2 e dc1.A variável dc1 é o valor absoluto da diferença de a2 e a1. Eu estou gerando um gráfico na matplotlib com os valores da variável a1 e gostaria as cores da 'colorbar' fossem definidos pelos valores da variável dc1. Mas não estou conseguindo fazer isso.

Abaixo segue o meu código.

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

a1 = np.array([1.5,2.5,2.0,5.5,7.0])

a2 = np.array([5.0,7.5,1.0,8,9])

dc1 = np.abs(a2-a1)

idx=np.arange(len(a1))

plt.scatter(idx,a1,s=dc1*100, c=dc1, cmap="BuPu",
edgecolors="black", linewidth=0.90)
plt.title(" MLP Keras")
plt.xlabel('Amostras\n')
plt.ylabel('Discrepâncias(m)')
plt.colorbar(dc1)
plt.grid(True)
plt.show()




Grato pela atenção

Continue reading...