Color Map на график рассеяния

Question

У меня есть три столбца в файле CSV, который звучит так:

X,Y, Cosine efficiency 
989.565,670.17,0.868684456 
660.941,-1502.48,0.597998523 
685.683,-1491.35,0.600137189 
-1454.88,-316.582,0.666894492 
-510.86,1016.79,0.870398677 
-1350.21,-788.027,0.61601682 
-944.034,-1342.79,0.565497692 
-966.098,-1327,0.56678693 
-1206.4,-1054.63,0.590095562 
1004.18,648.067,0.866589585 

Я могу читать значения X и Y и сделать график рассеяния следующим образом:

import csv 
# Reading data from csv file 
with open('my_file.csv') as csvfile: 
readCSV = csv.reader(csvfile, delimiter=',') 
    X = [] 
    Y = [] 
    C = [] 

for row in readCSV: 
    X_coordinates = row[0] 
    Y_coordinates = row[1] 
    cos_eff = row[2] 
    X.append(X_coordinates) 
    Y.append(Y_coordinates) 
    C.append(cos_eff) 
Xcoordinate = [float(X[c]) for c in range(1,len(X))] 
Ycoordinate=[float(Y[c]) for c in range(1,len(Y))] 
CosineEff=[float(C[c]) for c in range(1,len(C))] 

# Plotting the solar field 
plt.figure(1) 
plt.scatter(Xcoordinate,Ycoordinate, color = 'blue', s=2) 
plt.title("Solar field") 
plt.xlabel("Position, east-west [m]"); plt.ylabel("Position, east-west [m]") 
plt.show() 

Теперь я хочу сделать этот график рассеяния, используя цветовую карту, которая использует значения из третьего столбца.

Например, первый пункт: 989.565,670.17,0.868684456. Я уже построил координаты X и Y - в этом случае 989.565 и 670.17. Но я хочу использовать цветовую карту, чтобы поместить значение 0.868684456 на это построенное значение X & Y.

Я хочу использовать Brewer2mpl или pcolour, но я не знаю, с чего начать.

Я могу сделать цветную карту на Origin, используя функцию «сопоставленные цветом», но как я могу это сделать на Python?

Answer 1

plt.scatter(x_vals, y_vals, c = z_vals, cmap = 'rainbow') 

Вам нужно будет играть с добавлением s = и выбирая подходящий размер для очков. Там также есть красивый colormap set called cmocean, который вы можете захотеть установить