mirror of
https://github.com/wassname/geopandas.git
synced 2026-09-11 12:10:59 +08:00
@@ -253,8 +253,8 @@ class GeoDataFrame(GeoPandasBase, DataFrame):
|
||||
'id': str(i),
|
||||
'type': 'Feature',
|
||||
'properties':
|
||||
dict((k, v) for k, v in iteritems(row) if k != 'geometry'),
|
||||
'geometry': mapping(row['geometry']) }
|
||||
dict((k, v) for k, v in iteritems(row) if k != self._geometry_column_name),
|
||||
'geometry': mapping(row[self._geometry_column_name]) }
|
||||
|
||||
return json.dumps(
|
||||
{'type': 'FeatureCollection',
|
||||
|
||||
@@ -113,9 +113,6 @@ class GeoSeries(GeoPandasBase, Series):
|
||||
def __getitem__(self, key):
|
||||
return self._wrapped_pandas_method('__getitem__', key)
|
||||
|
||||
def __getslice__(self, i, j):
|
||||
return self._wrapped_pandas_method('__getslice__', i, j)
|
||||
|
||||
def order(self, *args, **kwargs):
|
||||
return self._wrapped_pandas_method('order', *args, **kwargs)
|
||||
|
||||
|
||||
+2
-1
@@ -28,7 +28,8 @@ def read_postgis(sql, con, geom_col='geom', crs=None, index_col=None,
|
||||
index_col, coerce_float, params
|
||||
|
||||
"""
|
||||
df = read_sql(sql, con, index_col, coerce_float, params)
|
||||
df = read_sql(sql, con, index_col=index_col, coerce_float=coerce_float,
|
||||
params=params)
|
||||
if geom_col not in df:
|
||||
raise ValueError("Query missing geometry column '{}'".format(
|
||||
geom_col))
|
||||
|
||||
@@ -188,6 +188,17 @@ class TestDataFrame(unittest.TestCase):
|
||||
self.assertTrue(data['type'] == 'FeatureCollection')
|
||||
self.assertTrue(len(data['features']) == 5)
|
||||
|
||||
def test_to_json_geom_col(self):
|
||||
df = self.df.copy()
|
||||
df['geom'] = df['geometry']
|
||||
df['geometry'] = np.arange(len(df))
|
||||
df.set_geometry('geom', inplace=True)
|
||||
|
||||
text = df.to_json()
|
||||
data = json.loads(text)
|
||||
self.assertTrue(data['type'] == 'FeatureCollection')
|
||||
self.assertTrue(len(data['features']) == 5)
|
||||
|
||||
def test_to_json_na(self):
|
||||
# Set a value as nan and make sure it's written
|
||||
self.df['Shape_Area'][self.df['BoroName']=='Queens'] = np.nan
|
||||
|
||||
Reference in New Issue
Block a user