Folium绘制独立PolyLine及按唯一ID着色问题求助

我看你现在遇到的核心问题是：把所有点位都塞进了单个LineString里，所以Folium会自动把所有点连成一条连续的线，这就是你看到的错误连接情况。要实现「不连接各线段」或者「按唯一ID给线段着色」，关键是要把每个独立的线段组（对应同一个唯一ID的点位）拆分成单独的Feature。

下面分两种场景给你具体的解决方案：

场景1：让线段互不连接（每个独立线段单独绘制） #

假设你的filteredData里有一个unique_id字段，用来标记哪些点属于同一个线段。我们需要按这个ID分组，每个组生成一个单独的LineString Feature：

import folium
import pandas as pd

# 假设filteredData是带unique_id的DataFrame，结构示例：
# filteredData = pd.DataFrame({
#     'lat': [36.8, 36.9, 37.0, 36.7, 36.6],
#     'lon': [-76.3, -76.2, -76.1, -76.4, -76.5],
#     'unique_id': [1, 1, 1, 2, 2]
# })

# 按unique_id分组，生成GeoJSON的FeatureCollection
features = []
for uid, group in filteredData.groupby('unique_id'):
    # 提取当前组的经纬度，注意LineString的坐标是[lon, lat]顺序
    coords = [[lon, lat] for lat, lon in zip(group['lat'], group['lon'])]
    features.append({
        'type': 'Feature',
        'geometry': {
            'type': 'LineString',
            'coordinates': coords
        },
        'properties': {
            'color': 'black'  # 默认黑色，后面场景2会改动态颜色
        }
    })

data = {
    'type': 'FeatureCollection',
    'features': features
}

# 创建地图并添加GeoJson
m = folium.Map(location=[36.862317, -76.3151], zoom_start=6)
folium.GeoJson(
    data,
    style_function=lambda x: {'color': x['properties']['color']}
).add_to(m)

m

这样每个unique_id对应的点会被连成单独的线段，不会和其他ID的线段连在一起。

场景2：按唯一ID给线段赋予不同颜色 #

如果想让不同ID的线段显示不同颜色，可以用颜色映射工具（比如matplotlib.cm）来生成唯一颜色，或者手动指定颜色列表：

方法1：用matplotlib生成动态颜色 #

import folium
import pandas as pd
import matplotlib.cm as cm
import matplotlib.colors as colors

# 获取所有唯一ID，生成颜色映射
unique_ids = filteredData['unique_id'].unique()
num_ids = len(unique_ids)
# 用viridis色阶生成对应数量的颜色
cmap = cm.get_cmap('viridis', num_ids)
color_map = {uid: colors.rgb2hex(cmap(i)[:3]) for i, uid in enumerate(unique_ids)}

# 按ID分组生成Feature
features = []
for uid, group in filteredData.groupby('unique_id'):
    coords = [[lon, lat] for lat, lon in zip(group['lat'], group['lon'])]
    features.append({
        'type': 'Feature',
        'geometry': {
            'type': 'LineString',
            'coordinates': coords
        },
        'properties': {
            'color': color_map[uid]
        }
    })

data = {'type': 'FeatureCollection', 'features': features}

m = folium.Map(location=[36.862317, -76.3151], zoom_start=6)
folium.GeoJson(
    data,
    style_function=lambda x: {'color': x['properties']['color'], 'weight': 2}  # 可以调整线宽
).add_to(m)

m

方法2：手动指定颜色列表 #

如果不想依赖matplotlib，也可以自己定义颜色列表，比如：

color_list = ['red', 'blue', 'green', 'orange', 'purple']
color_map = {uid: color_list[i % len(color_list)] for i, uid in enumerate(unique_ids)}

然后把这个color_map用到上面的Feature生成里就行。

额外提示：用PolyLine直接绘制（更直观） #

如果你觉得GeoJson的写法有点繁琐，也可以直接用folium.PolyLine逐个绘制每个线段：

m = folium.Map(location=[36.862317, -76.3151], zoom_start=6)

for uid, group in filteredData.groupby('unique_id'):
    # PolyLine的坐标是[lat, lon]顺序，和GeoJson相反哦！
    line_coords = [[lat, lon] for lat, lon in zip(group['lat'], group['lon'])]
    folium.PolyLine(
        locations=line_coords,
        color=color_map[uid],
        weight=2
    ).add_to(m)

m

这种方式更直观，适合快速调试。

内容的提问来源于stack exchange，提问作者id101112