'Polygon' 对象不可迭代- iPython 食谱

Question

我正在使用 Cartopy

学习 python 中的数据可视化

我有这段绘制非洲人口和 GDP 的代码。

def choropleth(ax, attr, cmap_name):
    # We need to normalize the values before we can
    # use the colormap.
    values = [c.attributes[attr] for c in africa]
    norm = Normalize(
        vmin=min(values), vmax=max(values))
    cmap = plt.cm.get_cmap(cmap_name)
    for c in africa:
        v = c.attributes[attr]
        sp = ShapelyFeature(c.geometry, crs,
                            edgecolor='k',
                            facecolor=cmap(norm(v)))
        ax.add_feature(sp)


fig, (ax1, ax2) = plt.subplots(
    1, 2, figsize=(10, 16),
    subplot_kw=dict(projection=crs))
draw_africa(ax1)
choropleth(ax1, 'POP_EST', 'Reds')
ax1.set_title('Population')

draw_africa(ax2)
choropleth(ax2, 'GDP_MD_EST', 'Blues')
ax2.set_title('GDP')

预期的输出应该是——

但我收到这样的错误 -

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-41-b443c58ecbd5> in <module>
      3     subplot_kw=dict(projection=crs))
      4 draw_africa(ax1)
----> 5 choropleth(ax1, 'POP_EST', 'Reds')
      6 ax1.set_title('Population')
      7 

<ipython-input-40-161126226479> in choropleth(ax, attr, cmap_name)
      8     for c in africa:
      9         v = c.attributes[attr]
---> 10         sp = ShapelyFeature(c.geometry, crs,
     11                             edgecolor='k',
     12                             facecolor=cmap(norm(v)))

~/anaconda3/lib/python3.8/site-packages/cartopy/feature/__init__.py in __init__(self, geometries, crs, **kwargs)
    219         """
    220         super(ShapelyFeature, self).__init__(crs, **kwargs)
--> 221         self._geoms = tuple(geometries)
    222 
    223     def geometries(self):

TypeError: 'Polygon' object is not iterable

我尝试在 github 上搜索此问题，但无济于事。谁能帮我解决这个问题？

这里是site供参考。

Answer 1

问题在于，代码试图将形状 Polygon 传递给需要 MultiPolygon 的函数。 swatchai 在这里的优雅解决方案是捕获 Polygons 并将它们放入列表中，以便将它们视为 MultiPolygons.

这是适合您的情况的代码：

for i, c in enumerate(africa):
    v = c.attributes[attr]
    print(i)
    # swatchai's Polygon catch logic
    if c.geometry.geom_type=='MultiPolygon':
        # this is a list of geometries
        sp = ShapelyFeature(c.geometry, crs,
                        edgecolor='k',
                        facecolor=cmap(norm(v)))
    elif c.geometry.geom_type=='Polygon': 
        # this is a single geometry
        sp = ShapelyFeature([c.geometry], crs,
                                edgecolor='k',
                                facecolor=cmap(norm(v)))   
    else:
        pass  #do not plot the geometry

'Polygon' 对象不可迭代- iPython 食谱

'Polygon' object is not iterable- iPython Cookbook

python

data-visualization

cartopy

jupyter-notebook