Блог

Я очень новичок в Matplotlib и python в целом и хочу показать точечный график с цветовой картой, например, hot, которая соответствует легенде.

Мои данные настроены в виде двух списков списков ( x_list и y_list ) длиной по 100 каждый.

 for i in x_list:
   x_list[i] = # list of data corresponding to a function where T = i
  

У меня есть такой график, где каждый цвет является значением T:

 for T in range(100):
    x = x_list[T]
    y = y_list[T]
    
    plt.scatter(x, y, label="T{}".format(T))
  

В идеале я хочу, чтобы это выглядело так из модуля seaborn, вот так

Где цвет, назначенный каждой точке, является градиентом цветовой карты в соответствии со значением T .

Могу ли я сделать это с помощью цветовой карты cm в Matplotlib?

Вы можете создать цветовой массив rgb и передать в качестве параметра функции разброса matplotlib

 colors = [(t/100,0,0) for t in range(len(x_list))] 
plt.scatter(x_list, y_list, color=colors)