着色大叶中的多边形

时间:2020-07-05 18:07:42

标签: polygon geojson folium

我想在地图上用folium在python中打印一些彩色的多边形,我需要一些帮助。我遇到的问题是我所有的多边形只能得到一种颜色。这是我循环运行的代码的一部分:

import folium
from shapely.geometry import Polygon

def map_color_rsrp(rsrp):
    if (int(rsrp) in range(-70, -50)):
        return '#10FF00'
    elif (int(rsrp) in range(-90, -70)):
        return 'green'
    elif (int(rsrp) in range(-110, -90)):
        return 'blue'
    elif (int(rsrp) in range(-130, -110)):
        return '#FF7000'
    else:  # range(-150, -130)
        return 'grey'

# create map
grid_map = folium.Map((51.443787, 7.262206), zoom_start=14)

# create FeatureGroups for RSRP & RSRQ
rsrp_grid_fg = folium.FeatureGroup(name="RSRP", overlay=True)

grid_poly_rows = [
    [-78.2,  [(7.251043000000002, 51.444325000000035), (7.251043000000002, 51.44462500000004), (7.251343000000002, 51.44462500000004), (7.251343000000002, 51.444325000000035)]],
    [-89.3,  [(7.251343000000002, 51.444325000000035), (7.251343000000002, 51.44462500000004), (7.251643000000002, 51.44462500000004), (7.251643000000002, 51.444325000000035)]],
    [-113.7, [(7.251643000000002, 51.44402500000003), (7.251643000000002, 51.444325000000035), (7.2519430000000025, 51.444325000000035), (7.2519430000000025, 51.44402500000003)]],
    [-112.3, [(7.251643000000002, 51.444325000000035), (7.251643000000002, 51.44462500000004), (7.2519430000000025, 51.44462500000004), (7.2519430000000025, 51.444325000000035)]],
    [-133.7, [(7.2519430000000025, 51.44402500000003), (7.2519430000000025, 51.444325000000035), (7.252243000000003, 51.444325000000035), (7.252243000000003, 51.44402500000003)]]
]

for row in grid_poly_rows:
    mean_rsrp = row[0]
    rsrp_tooltip_str = str(mean_rsrp)
    rsrp_color = map_color_rsrp(mean_rsrp)
    style_ = {
        'fillColor': rsrp_color,
        'color': rsrp_color,
        'weight': 1,
        'fillOpacity': 0.5}
    folium.GeoJson(Polygon(row[1]),
                   style_function=lambda x: style_,
                   tooltip=rsrp_tooltip_str
                   ).add_to(rsrp_grid_fg)

rsrp_grid_fg.add_to(grid_map)

# add LayerControl
folium.LayerControl(collapsed=False).add_to(grid_map)

grid_map.save(outfile="grid.html")

我认为style_function出了点问题。任何建议表示赞赏!预先感谢!

干杯ninjab3s

1 个答案:

答案 0 :(得分:0)

我成功了!我发现了这篇帖子https://stackoverflow.com/a/53816162/13872164,该帖子解释了lambda样式函数不会立即执行。 以下更改使我的代码正常工作:

df.infer_objects().dtypes

worker    object
day        int64
tasks     object
dtype: object

下一次我要在发帖之前深入挖掘