Вернуть ZipCode, если точка в многоугольнике (стройная)

Question

У меня есть DataFrame A с колонкой точек:

   points                      Code
0  Point(1.23, 1.34)            ?
1  Point(1.32, 3.56)            ?
2  Point(-1.09, 2.11)           ?
.
.

У меня также есть другой DataFrame B со столбцом Polygon:

   Code   Polygon
0  123    Polygon((-2,3),(1,4),(3,3),(-1,-2))
1  203    Polygon((-1,2),(0,2),(4,1),(-2,-1))
.
.

Как я могу передать код от B до A, если точка находится в пределах Polygon?

Answer 1

Предположим, у вас есть два кадра данных:

df1 = GeoDataFrame(
    [[Point(1.23, 1.34)],
    [Point(1.32, 3.56)],
    [Point(-1.09, 2.11)]],
    columns=['geometry'],  
    geometry='geometry')

df2 = GeoDataFrame(
    [[Polygon([(-2,3),(1,4),(3,3),(-1,-2)]), 123],
    [Polygon([(-1,2),(0,2),(4,1),(-2,-1)]), 203]],
    columns=['geometry', 'Code'],  
    geometry='geometry')

Вы можете сделать это вручную:

# add the Code column:
df1.loc[:,'Code'] = [None]*len(df1.geometry)
# compute the intersections
for i, p in enumerate(df1.geometry):
    for j, pol in enumerate(df2.geometry):
        if pol.contains(p):
            df1['Code'][i] = df2['Code'][j]
            break

Или вы можете сделать это с помощью пространственного соединения:

df1 = gpd.sjoin(df1, df2, how="inner", op='intersects')
# remove the index_right extra column that is generated:
df1.drop('index_right', axis=1, inplace=True)

Осторожно, второй метод будет повторять строки точек, если точка пересекает более одного столбца, первый не будет.