CODEWITHSUNDEEP
pythonpandasdataframe
roudan
roudan

Reputation: 4208

filter a dataframe with rows inside a polygon

I have a dataframe with columns of x, y, data. and I have a list for polygon coordinate like this: polygon=[x1,y1,x2,y2,x3,y3,x4,y4]

I'd like to filter out all the rows of dataframe with (x,y) outside the polygon


df.columns=['x','y','data']
polygon=[x1,y1,x2,y2,x3,y3,x4,y4]

df_1= df inside polygon

How to implement the last line? Thanks

Upvotes: 0

Views: 685

Answers (1)

tozCSS
tozCSS

Reputation: 6114

See Shapely:

from shapely.geometry import Point
from shapely.geometry.polygon import Polygon

df['point'] = df.apply(lambda row: Point(row['x'],row['y']),axis=1)
polygon = Polygon([(x1,y1), (x2,y2), (x3,y3), (x4,y4)])
df_1 = df[df['point'].apply(polygon.contains)].copy()

Upvotes: 1

Related Questions