Question

以下代码生成一个网络地图，其中国家/地区的颜色来自world.json。

import folium

map=folium.Map(location=[30,30],tiles='Stamen Terrain')

map.add_child(folium.GeoJson(data=open('world.json', encoding='utf-8-sig'),
name="Unemployment",
style_function=lambda x: {'fillColor':'green' if x['properties']['POP2005'] <= 10000000 else 'orange' if 10000000 < x['properties']['POP2005'] < 20000000 else 'red'}))

map.save('file.html')

world.json的链接。

我想知道是否可以使用def而不是lambda函数创建的普通函数作为style_function参数的值。我尝试为此创建一个函数：

def feature(x):
    file = open("world.json", encoding='utf-8-sig')
    data = json.load(file)
    population = data['features'][x]['properties']['POP2005']
    d ={'fillColor':'green' if x['properties']['POP2005'] <= 10000000 else 'orange' if 10000000 < x['properties']['POP2005'] < 20000000 else 'red'}
    return d

但是，我无法考虑如何在style_function中使用它。这是可能的还是lambda函数在这里是不可替代的？

Answer 1

style_function lambda可以替换为这样的函数：

def style_function(x):
    return {'fillColor':'green' if x['properties']['POP2005'] <= 10000000 else 'orange' if 10000000 < x['properties']['POP2005'] < 20000000 else 'red'}))

然后你可以将函数名称传递给kwarg：

folium.GeoJson(
    data=...,
    name=...,
    style_function=style_function
)

Answer 2

如果我理解正确，你需要制作这样的函数（x是geojson）：

def my_style_function(x):
    color = ''
    if x['properties']['POP2005'] <= 10e6:
        color = 'green'
    elif x['properties']['POP2005'] < 2*10e6:
        color = 'orange'
    return {'fillColor': color if color else 'red'}

并简单地将其分配给style_function参数（不带括号）：

map.add_child(folium.GeoJson(data=open('world.json', encoding='utf-8-sig'),
                             name="Unemployment",
                             style_function=my_style_function))

正常＆＃39; def＆＃39;功能而不是lambda

2 个答案: