2016-03-28 6 views
-1

ich drei Spalten in einer CSV-Datei haben, die wie folgt lautet:Farbkarte auf Streudiagramm

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 

ich die X- und Y-Werte lesen kann und ein Streudiagramm auf folgende Weise machen:

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() 

Nun möchte ich dieses Streudiagramm mit einer Farbkarte erstellen, die die Werte aus der dritten Spalte verwendet.

Zum Beispiel ist der erste Punkt: 989.565.670,17,0.868684456. Ich habe bereits die X- und Y-Koordinaten gezeichnet - in diesem Fall 989.565 und 670.17. Aber ich möchte eine Farbkarte verwenden, um den Wert von 0,868684456 auf diesen aufgetragenen X & Y-Wert zu setzen.

Ich möchte Brewer2mpl oder pcolour verwenden, aber ich weiß nicht, wo ich anfangen soll.

Ich kann eine Farbkarte auf Origin mit der Funktion "Color Mapping" erstellen, aber wie kann ich das auf Python machen?

Antwort

2

Sie werden mit dem Hinzufügen von s = und der Auswahl einer geeigneten Größe für die Punkte spielen wollen. Es gibt auch eine schöne colormap set called cmocean, die Sie möglicherweise installieren möchten

+0

Vielen Dank! Es klappt! Ich schaue mir cmocean an! –