@@ -420,19 +420,18 @@ def read_raster(file_name, band=[1], src_crs=None, window=False, geometry=False,
420420 intensity = inten [band_idx , :]
421421 return meta , intensity .reshape ((len (band ), meta ['height' ]* meta ['width' ]))
422422
423- def read_vector (file_name , inten_name = [ 'intensity' ] , dst_crs = None ):
424- """ Read vector file format supported by fiona. Each intensity name is
423+ def read_vector (file_name , field_name , dst_crs = None ):
424+ """ Read vector file format supported by fiona. Each field_name name is
425425 considered an event.
426426
427427 Parameters:
428428 file_name (str): vector file with format supported by fiona and
429429 'geometry' field.
430- inten_name (list(str)): list of names of the columns of the
431- intensity of each event.
430+ field_name (list(str)): list of names of the columns with values.
432431 dst_crs (crs, optional): reproject to given crs
433432
434433 Returns:
435- np.array (lat), np.array (lon), geometry (GeiSeries), np.array (intensity )
434+ np.array (lat), np.array (lon), geometry (GeiSeries), np.array (value )
436435 """
437436 LOGGER .info ('Reading %s' , file_name )
438437 data_frame = gpd .read_file (file_name )
@@ -443,10 +442,10 @@ def read_vector(file_name, inten_name=['intensity'], dst_crs=None):
443442 else :
444443 geometry = data_frame .geometry .to_crs (dst_crs )
445444 lat , lon = geometry [:].y .values , geometry [:].x .values
446- intensity = np .zeros ([len (inten_name ), lat .size ])
447- for i_inten , inten in enumerate (inten_name ):
448- intensity [i_inten , :] = data_frame [inten ].values
449- return lat , lon , geometry , intensity
445+ value = np .zeros ([len (field_name ), lat .size ])
446+ for i_inten , inten in enumerate (field_name ):
447+ value [i_inten , :] = data_frame [inten ].values
448+ return lat , lon , geometry , value
450449
451450def write_raster (file_name , data_matrix , meta ):
452451 """ Write raster in GeoTiff format
0 commit comments