2016-03-15 32 views
5

Sto provando a creare una barra colori Matplotlib su GeoPandas.Colorbar on Geopandas

import geopandas as gp 
import pandas as pd 
import matplotlib.pyplot as plt 

#Import csv data 
df = df.from_csv('data.csv') 

#Convert Pandas DataFrame to GeoPandas DataFrame 
g_df = g.GeoDataFrame(df) 

#Plot 
plt.figure(figsize=(15,15)) 
g_plot = g_df.plot(column='column_name',colormap='hot',alpha=0.08) 
plt.colorbar(g_plot) 

ottengo il seguente errore:

AttributeError       Traceback (most recent call last) 
<ipython-input-55-5f33ecf73ac9> in <module>() 
     2 plt.figure(figsize=(15,15)) 
     3 g_plot = g_df.plot(column = 'column_name', colormap='hot', alpha=0.08) 
----> 4 plt.colorbar(g_plot) 

... 

AttributeError: 'AxesSubplot' object has no attribute 'autoscale_None' 

io non sono sicuro di come ottenere colorbar al lavoro.

+1

GeoDataFrame restituisce un oggetto 'Axes' matplotlib. 'plt.colorbar' richiede un oggetto mappabile (un' Axes' non è un oggetto mappabile). – tom

risposta

10

C'è un PR di aggiungere questo al geoapandas (https://github.com/geopandas/geopandas/pull/172), ma per ora, è possibile aggiungere da soli con questa soluzione:

## make up some random data 
df = pd.DataFrame(np.random.randn(20,3), columns=['x', 'y', 'val']) 
df['geometry'] = df.apply(lambda row: shapely.geometry.Point(row.x, row.y), axis=1) 
gdf = gpd.GeoDataFrame(df) 

## the plotting 

vmin, vmax = -1, 1 

ax = gdf.plot(column='val', colormap='hot', vmin=vmin, vmax=vmax) 

# add colorbar 
fig = ax.get_figure() 
cax = fig.add_axes([0.9, 0.1, 0.03, 0.8]) 
sm = plt.cm.ScalarMappable(cmap='hot', norm=plt.Normalize(vmin=vmin, vmax=vmax)) 
# fake up the array of the scalar mappable. Urgh... 
sm._A = [] 
fig.colorbar(sm, cax=cax) 

La soluzione viene da Matplotlib - add colorbar to a sequence of line plots. E la ragione per cui devi fornire vmin e vmax da solo è perché la barra di colore non viene aggiunta in base ai dati stessi, quindi devi indicare quale dovrebbe essere il collegamento tra valori e colore.