Применение пользовательских пространственных операций

Работа с геопространственными данными в Python

Joris Van den Bossche

Open source software developer and teacher, GeoPandas maintainer

Суммарная длина рек в радиусе 50 км от каждого города?

Для одной точки (cairo):

area = cairo.buffer(50000)

rivers_within_area = rivers.intersection(area)

print(rivers_within_area.length.sum() / 1000)

186.397219642

Метод apply()

Series.apply(): вызывает функцию для каждого значения Series

Series.apply(function, **kwargs)

function: функция, вызываемая для каждого значения; значение передаётся первым аргументом
**kwargs: дополнительные аргументы, передаваемые в функцию

Для GeoSeries функция вызывается как function(geom, **kwargs) для каждого geom в GeoSeries

Применение пользовательской пространственной операции

Применяемая функция:

def river_length(geom, rivers):
    area = geom.buffer(50000)
    rivers_within_area = rivers.intersection(area)
    return rivers_within_area.length.sum() / 1000

Вызов функции для одной геометрии:

river_length(cairo, rivers=rivers)

186.3972196423455

Применение ко всем городам:

cities.geometry.apply(river_length, rivers=rivers)

Применение пользовательской пространственной операции

Применение ко всем городам:

cities.geometry.apply(river_length, rivers=rivers)

0        0.000000
1        0.000000
2      106.072198
          ...

Применение пользовательской пространственной операции

Применение ко всем городам с сохранением результата в новый столбец:

cities['river_length'] = cities.geometry.apply(river_length, rivers=rivers)
cities.head()

             name                       geometry  river_length
0    Vatican City    POINT (1386304.6 5146502.5)      0.000000
1      San Marino    POINT (1385011.5 5455558.1)      0.000000
2           Vaduz    POINT (1059390.7 5963928.5)    106.072198
..            ...                            ...           ...

Давайте потренируемся!

Работа с геопространственными данными в Python