Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
dship2postgis / geometry_generator.py
Size: Mime:
# -*- coding: utf-8 -*-
import geopandas as gpd
from geoalchemy2 import WKTElement
from shapely.geometry import Point

from dshipparser.parser import LAT_COL
from dshipparser.parser import LON_COL

from dship2postgis.config import Config as cfg


def _make_geodataframe(df):
    """
    Creates a new GeoDataFrame from passed pandas Dataframe
    :param df:
    :return:
    """
    geometry = [Point(xy) for xy in zip(df[LON_COL], df[LAT_COL])]
    gdf = gpd.GeoDataFrame(df, geometry=geometry, crs="EPSG:4326")
    gdf['geom'] = gdf['geometry'].apply(lambda x: WKTElement(x.wkt, srid=4326))

    return gdf