使用geopandas从pyplot检索数据

时间:2019-05-02 10:33:54

标签: python python-3.x matplotlib gis geopandas

我试图避免将matplotlib图形保存到文件中,只是为了调用它并为张量流模型预处理图像。我看到this使我感到乐观,不幸的是,geopandas具有内置的绘图功能。我想知道是否有人有什么想法?

理想情况下,我想要一个包含所有点的强度值的numPy图像数组

data = gp.read_file('/path_to.shp')

fig = plt.figure()
ax = fig.add_subplot(111)

lines = data.plot(ax=ax)
print(lines[0].get_data())

不幸的是,我得到了错误:

Traceback (most recent call last):
  File "GIS2img.py", line 108, in <module>
    print(lines[0].get_data())
TypeError: 'AxesSubplot' object does not support indexing

我想为每个图形获取数组中的RGB图像数据

感谢您的帮助

编辑:澄清的问题

1 个答案:

答案 0 :(得分:0)

解决了!

感谢ImportanceOfBeingErnest的帮助。

您可以使用fig.canvas.draw()绘制到画布上,然后使用np.array(fig.canvas.renderer._renderer)检索该图像。