python - folium 中线条的动态样式

Question

我一直在尝试理解来自 folium 的 TimestampedGeoJson 插件。

我想绘制随时间改变颜色的线条。目前，我所做的是每次需要更改颜色时完全重画一条线，这会带来巨大的开销。

另一个问题是如何在特征中指定时间。目前，我有这个例子:

import folium
from folium.plugins import TimestampedGeoJson

m = folium.Map(
    location=[42.80491692, -4.62577249],
    zoom_start=10
)

data = [
{
    'coordinates': [
        [-4.018876661, 43.11843382],
        [-4.856537491, 42.82202193],
    ],
    'dates': [
        '2017-06-02T00:00:00',
        '2017-06-02T00:10:00'
    ],
    'color': 'red'
},
{
    'coordinates': [
        [-4.018876661, 43.11843382],
        [-4.856537491, 42.82202193],
    ],
    'dates': [
        '2017-06-02T00:00:00',
        '2017-06-02T00:20:00'
    ],
    'color': 'blue'
},
]

features = [
    {
        'type': 'Feature',
        'geometry': {
            'type': 'LineString',
            'coordinates': d['coordinates'],
        },
        'properties': {
            'times': d['dates'],
            'style': {
                'color': d['color'],
                'weight': d['weight'] if 'weight' in d else 5
            }
        }
    }
    for d in data
]

TimestampedGeoJson({
    'type': 'FeatureCollection',
    'features': features,
}, period='PT1M', add_last_point=True).add_to(m)

m.save('dynamic4.html')

对我来说，第一个日期没有任何意义，但显然它是必需的，否则浏览器将不会绘制任何东西。

所以:

a) 如何在不重新绘制线条的情况下更改样式？b) 时间是什么意思？如何指定一致的时间顺序？

Answer 1

我将首先尝试单独解决您的问题，然后我会提供一个完整的解决方案来说明我最终会做什么。本质上:

1. 更改 TimestampedGeoJson _template 变量以更改 style_function，这将使样式动态化
2. 确保在 TimestampedGeoJson 数据中每个坐标都有一个时间步
3. 为避免混淆，尽量不要重叠特征或特征在特定时间步长内缺失数据
4. 我相信在您的场景中您只有一个特征，但会在不同的时间步改变颜色

解决您的问题:

a) 如何在不重新绘制线条的情况下更改样式？

我认为从 folium 本身不可能，有必要将 style_function 传递给 TimestampedGeoJson，目前它甚至不是类 init 的参数。这似乎很难做到，因为您需要将 python style_function 转换为 javascript style_function。

将有一个简单的解决方法。 class definition of TimestampedGeoJson里面它使用 _template 变量作为 javascript 代码的字符串模板，因此您可以根据需要调整此模板，但使用 javascript。

class TimestampedGeoJson(MacroElement):
         .... hidding lines to save space
    _template = Template("""
        {% macro script(this, kwargs) %}
                 .... hidding lines to save space
                    style: function (feature) {
                        return feature.properties.style;
                    },
                    onEachFeature: function(feature, layer) {
                        if (feature.properties.popup) {
                        layer.bindPopup(feature.properties.popup);
                        }
                    }
                })
        {% endmacro %}
        """)  # noqa
     ..... hidding lines to save space

因此，为了在每个时间步更改线条颜色，您可以更改模板的这一部分:

style: function (feature) {
                        return feature.properties.style;
                    },

通过这个:遍历一组颜色

                    style: function(feature) {
                        lastIdx=feature.properties.colors.length-1
                        currIdx=feature.properties.colors.indexOf(feature.properties.color);
                        if(currIdx==lastIdx){
                            feature.properties.color = feature.properties.colors[0]  
                        }
                        else{
                            feature.properties.color =feature.properties.colors[currIdx+1] 
                        }
                        return {color: feature.properties.color}
                    },

更改它以便在每个时间步更新 properties.style 中的颜色。

b) 时间是什么意思？如何指定一致的时间顺序？

TimestampedGeoJson 正在使用库 Leaflet.TimeDimension，因此 TimestampedGeoJson 对应于 L.TimeDimension.Layer.GeoJSON .

从你得到的文档中

“coordTimes、times 或 linestringTimestamps:可以与几何关联的时间数组(日期字符串或 ms)。在 LineString 的情况下，它必须具有与 LineString 中的坐标一样多的项目。 ”

所以本质上要保持一致，只需确保 1.对于每个特征，时间大小与坐标相同，并且 2. 使用有效的日期字符串或 ms 格式 3. 如果您的日期增加一个常数，请将您的期间设置为该值

放在一起我主要改变了你之前的例子:1) 使用新的 style_function 添加了一个_template 变量并更改了 TimestampedGeoJson 默认模板

2) 稍微更改了两个特征坐标，以表明您设置的两个特征是重叠的，并且有一些时间步长，在某些时间步长中，只定义了第一个特征，后来只定义了第二个特征，所以它会混淆正在发生的事情在每个时间步。

3) 为每个特征添加了一个颜色列表来循环

from jinja2 import Template
_template = Template("""
    {% macro script(this, kwargs) %}
        L.Control.TimeDimensionCustom = L.Control.TimeDimension.extend({
            _getDisplayDateFormat: function(date){
                var newdate = new moment(date);
                console.log(newdate)
                return newdate.format("{{this.date_options}}");
            }
        });
        {{this._parent.get_name()}}.timeDimension = L.timeDimension(
            {
                period: {{ this.period|tojson }},
            }
        );
        var timeDimensionControl = new L.Control.TimeDimensionCustom(
            {{ this.options|tojson }}
        );
        {{this._parent.get_name()}}.addControl(this.timeDimensionControl);
        var geoJsonLayer = L.geoJson({{this.data}}, {
                pointToLayer: function (feature, latLng) {
                    if (feature.properties.icon == 'marker') {
                        if(feature.properties.iconstyle){
                            return new L.Marker(latLng, {
                                icon: L.icon(feature.properties.iconstyle)});
                        }
                        //else
                        return new L.Marker(latLng);
                    }
                    if (feature.properties.icon == 'circle') {
                        if (feature.properties.iconstyle) {
                            return new L.circleMarker(latLng, feature.properties.iconstyle)
                            };
                        //else
                        return new L.circleMarker(latLng);
                    }
                    //else
                    return new L.Marker(latLng);
                },
                style: function(feature) {
                    lastIdx=feature.properties.colors.length-1
                    currIdx=feature.properties.colors.indexOf(feature.properties.color);
                    if(currIdx==lastIdx){
                        feature.properties.color = feature.properties.colors[currIdx+1] 
                    }
                    else{
                        feature.properties.color =feature.properties.colors[currIdx+1] 
                    }
                    return {color: feature.properties.color}
                },
                onEachFeature: function(feature, layer) {
                    if (feature.properties.popup) {
                    layer.bindPopup(feature.properties.popup);
                    }
                }
            })
        var {{this.get_name()}} = L.timeDimension.layer.geoJson(
            geoJsonLayer,
            {
                updateTimeDimension: true,
                addlastPoint: {{ this.add_last_point|tojson }},
                duration: {{ this.duration }},
            }
        ).addTo({{this._parent.get_name()}});
    {% endmacro %}
    """)
import folium
from folium.plugins import TimestampedGeoJson

m = folium.Map(
    location=[42.80491692, -4.62577249],
    zoom_start=9
)

data = [
{
    'coordinates': [
        [-4.018876661, 43.11843382],
        [-4.856537491, 42.82202193],
    ],
    'dates': [
        '2017-06-02T00:00:00',
        '2017-06-02T00:10:00'
    ],
    'color': 'brown',
    'colors':["black","orange","pink"],
},
{
    'coordinates': [
        [-4.058876661, 43.11843382],
        [-4.936537491, 42.82202193],
    ],
    'dates': [
        '2017-06-02T00:00:00',
        '2017-06-02T00:10:00'
    ],
    'color': 'blue',
    'colors':["red","yellow","green"],
},
]

features = [
    {
        'type': 'Feature',
        'geometry': {
            'type': 'LineString',
            'coordinates': d['coordinates'],
        },
        'properties': {
            'times': d['dates'],
            'color': d["color"],
            'colors':d["colors"]
        }
    }
    for d in data
]

t=TimestampedGeoJson({
    'type': 'FeatureCollection',
    'features': features,
}, period='PT10H', add_last_point=True)
t._template=_template
t.add_to(m)

m.save('original.html')