Question

我的python代码中存在一个循环，该循环将功能添加到了我的叶片图。每个功能都保存在字典（称为产品）中，每个功能都具有参数字典（GeoJson字符串和属性）。我想根据其中一个参数为功能着色。

我首先为参数创建一个唯一值列表。然后将颜色映射到每个唯一值：

orb_list = [value['relativeorbitnumber'] for key, value in products.items()]
orb_list = list(set(orb_list))

color_cycle = cycler(color=['#8e0038', '#8e0073', '#59008e'], fillColor=['#8e0038', '#8e0073', '#59008e'])

orb_colors=dict([[i, sty] for i, sty in zip(orb_list, cycle(color_cycle))])

接下来，我创建一个循环，在其中检索与特征参数值相对应的color和fillColor，并在style_function中使用它：

for key,value in products.items():
    footprint = json.dumps(wkt.loads(products[key]['footprint']))
    fillColor = orb_colors[products[key]['relativeorbitnumber']]['fillColor']
    color = orb_colors[products[key]['relativeorbitnumber']]['color']
    feat = folium.GeoJson(footprint,
               style_function=lambda x: {'fillColor':fillColor,'color':color},
               highlight_function=lambda feature: {'fillcolor':'green','color':'green'},
               name='Footprint')

    feat.add_to(mapa)

我以为这可以用，但是所有功能似乎都由循环中的最后“ fillColor”和“ color”来着色。谁能解释为什么？我假定直到所有循环完成后才渲染要素，因此“ color”和“ fillColor”的最后一个值用于所有要素。您能建议如何修改我的代码以解决此问题吗？还是我会以错误的方式进行操作，您可以建议一种更清洁的方法？

Answer 1

您遇到了Python closures

的常见陷阱

您传递给例如style_function不会立即在循环中执行，而是稍后执行。那时，将从外部作用域中检索fillColor（因为它没有在lambda表达式创建的内部作用域中定义），此时它将具有最后一个值。

演示正在发生的事情：

In [1]: out = []
   ...: for k in ('a', 'b', 'c'):
   ...:     l = lambda: k
   ...:     out.append(l)
   ...: [l() for l in out]
Out[1]: ['c', 'c', 'c']

为防止这种情况，您可以将变量的（当前）值作为default argument传递给由lambda表达式创建的函数，以便在稍后执行该函数时使用该值：

In [2]: out = []
   ...: for k in ('a', 'b', 'c'):
   ...:     l = lambda k=k: k
   ...:     out.append(l)
   ...: [l() for l in out]
Out[2]: ['a', 'b', 'c']

或者在您的代码中：

for key, value in products.items():
    footprint = json.dumps(wkt.loads(products[key]["footprint"]))
    fillColor = orb_colors[products[key]["relativeorbitnumber"]]["fillColor"]
    color = orb_colors[products[key]["relativeorbitnumber"]]["color"]
    feat = folium.GeoJson(
        footprint,
        style_function=lambda x, fillColor=fillColor, color=color: {
            "fillColor": fillColor,
            "color": color,
        },
        highlight_function=lambda feature: {"fillcolor": "green", "color": "green"},
        name="Footprint",
    )

    feat.add_to(mapa)

这应该可以解决问题。

Folium：将GeoJson添加到地图时的颜色映射

1 个答案: