Geopandas上的Colorbar

Question

我想在GeoPandas上创建一个Matplotlib颜色条.

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)

我收到以下错误:

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'

我不知道如何让colorbar工作.

Answer 1

编辑:下面引用的PR已合并到geopandas master中.现在你可以简单地做:

gdf.plot(column='val', cmap='hot', legend=True)

并自动添加颜色条.

笔记:

• legend=True 告诉Geopandas添加颜色条.
• colormap现在被称为cmap.
• vmin并且vmax不再需要了.

有一个PR将其添加到geoapandas(https://github.com/geopandas/geopandas/pull/172),但是现在,您可以使用此解决方法自行添加它:

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

解决方法来自Matplotlib - 将颜色条添加到一系列线图中.你必须提供vmin和vmax你自己的原因是因为没有根据数据本身添加颜色条,因此你必须指示值和颜色之间的联系应该是什么.