@@ -971,7 +971,7 @@ def _read_bodc_data(fn_bodc, date_start=None, date_end=None, header_length: int
971
971
### ~ Plotting ~ ###
972
972
##############################################################################
973
973
974
- def plot_timeseries (self , id , var_list = ["ssh" ], date_start = None , date_end = None , plot_line = False ):
974
+ def plot_timeseries (self , var_list = ["ssh" ], date_start = None , date_end = None , plot_line = False ):
975
975
"""
976
976
Quick plot of time series stored within object's dataset
977
977
Parameters
@@ -992,9 +992,10 @@ def plot_timeseries(self, id, var_list=["ssh"], date_start=None, date_end=None,
992
992
var_list = [var_list ]
993
993
994
994
for var_str in var_list :
995
- dim_str = self .dataset [var_str ].dims [0 ]
996
- x = np .array (self .dataset [dim_str ])
997
- y = np .array (self .dataset [var_str ])
995
+ # dim_str = self.dataset[var_str].dims[0] # commented out: could pull wrong dimensio
996
+ # x = np.array(self.dataset[dim_str]) # return indices, not coordinate values
997
+ x = np .array (self .dataset ["time" ])
998
+ y = np .array (self .dataset [var_str ].squeeze ())
998
999
999
1000
# Use only values between stated dates
1000
1001
start_index = 0
@@ -1019,7 +1020,20 @@ def plot_timeseries(self, id, var_list=["ssh"], date_start=None, date_end=None,
1019
1020
plt .legend (var_list )
1020
1021
# Title and axes
1021
1022
plt .xlabel ("Date" )
1022
- plt .title ("Site: " + self .dataset .site_name )
1023
+
1024
+ try :
1025
+ self .dataset .id_name
1026
+ except AttributeError :
1027
+ try :
1028
+ self .dataset .site_name
1029
+ except AttributeError :
1030
+ title_str = ""
1031
+ else :
1032
+ title_str = f"Site: { self .dataset .site_name } "
1033
+ else :
1034
+ title_str = f"Site: { self .dataset .id_name } "
1035
+
1036
+ plt .title (title_str )
1023
1037
1024
1038
return fig , ax
1025
1039
@@ -1057,19 +1071,22 @@ def plot_on_map_multiple(cls, tidegauge_list, color_var_str=None):
1057
1071
Y = []
1058
1072
C = []
1059
1073
for tg in tidegauge_list :
1060
- X .append (tg .dataset .longitude )
1061
- Y .append (tg .dataset .latitude )
1074
+ X .append (tg .dataset .longitude . values )
1075
+ Y .append (tg .dataset .latitude . values )
1062
1076
if color_var_str is not None :
1063
1077
C .append (tg .dataset [color_var_str ].values )
1064
1078
1079
+ X = np .hstack (X )
1080
+ Y = np .hstack (Y )
1065
1081
title = ""
1066
1082
if color_var_str is None :
1067
1083
fig , ax = plot_util .geo_scatter (X , Y , title = title )
1068
1084
else :
1085
+ C = np .hstack (C )
1069
1086
fig , ax = plot_util .geo_scatter (X , Y , title = title , c = C )
1070
1087
1071
- ax .set_xlim ((min (X ) - 10 , max (X ) + 10 ))
1072
- ax .set_ylim ((min (Y ) - 10 , max (Y ) + 10 ))
1088
+ ax .set_xlim ((np . min (X ) - 10 , np . max (X ) + 10 ))
1089
+ ax .set_ylim ((np . min (Y ) - 10 , np . max (Y ) + 10 ))
1073
1090
return fig , ax
1074
1091
1075
1092
##############################################################################
0 commit comments