Seaborn Heatmap Mit

Question

Ich laufe in einige Probleme Heatmap in Seaborn

Code verwendet:

df[['latitude', 'longitude', 'interest_level']] 
a = df.pivot('latitude', 'longitude', 'interest_level') 

Fehler:

ValueError: Index contains duplicate entries, cannot reshape 

Zinsniveau hat mehrere doppelte Werte. Was ich mir vorstelle, ist, dass das lat/long ein 2D-Diagramm bilden würde, wobei das Zinsniveau der Wert in der Karte wäre. Das Zinsniveau ist kategorisch mit nur 3 eindeutigen Werten.

bin ich die falsche Klasse verwenden, wie ich bin nach dem Seaborn Heatmap Tutorial http://seaborn.pydata.org/generated/seaborn.heatmap.html

Answer 1

Sie benötigen pivot_table mit einigen Aggregatfunktion wie mean, sum ...:

#subset for pivot_table or groupby solution is not necessary, you can omit it 
#df = df[['latitude', 'longitude', 'interest_level']] 
a = df.pivot_table(index='latitude', 
        columns='longitude', 
        values='interest_level', 
        aggfunc='mean') 

Oder groupby, aggregieren Funktion und unstack:

a = df.groupby(['latitude','longitude'])['interest_level'].mean().unstack() 

Samp le:

df = pd.DataFrame({'latitude':[53,54,55,55], 
        'longitude':[10,11,12,12], 
        'interest_level':[1,5,2,6], 
        'another_col':[4,7,4,2]}) 
print (df) 
    another_col interest_level latitude longitude 
0   4    1  53   10 
1   7    5  54   11 
2   4    2  55   12 <-duplicates for 55,12 
3   2    6  55   12 <-duplicates for 55,12 

a = df.pivot_table(index='latitude', 
        columns='longitude', 
        values='interest_level', 
        aggfunc='mean') 
print (a) 
longitude 10 11 12 
latitude     
53   1.0 NaN NaN 
54   NaN 5.0 NaN 
55   NaN NaN 4.0 <- (2+6)/2 = 4 

Last:

ax = sns.heatmap(a)