@@ -420,7 +420,7 @@ def cess_date_from_rain(
420
420
of soil moisture less than `dry_thresh` .
421
421
Soil moisture is estimated through a simple water balance model
422
422
driven by rainfall `daily_rain` , evapotranspiration `et`, soil capacity `taw`
423
- and initislized at soil moisture value `sminit` .
423
+ and initialized at soil moisture value `sminit` .
424
424
425
425
Parameters
426
426
----------
@@ -763,13 +763,13 @@ def seasonal_onset_date(
763
763
end_month = first_end_date .dt .month .values
764
764
765
765
# Apply daily grouping by season
766
- grouped_daily_data = daily_tobegroupedby_season (
766
+ seasonally_labeled_daily_data = daily_tobegroupedby_season (
767
767
daily_rain , search_start_day , search_start_month , end_day , end_month
768
768
)
769
769
# Apply onset_date
770
770
seasonal_data = (
771
- grouped_daily_data [daily_rain .name ]
772
- .groupby (grouped_daily_data ["seasons_starts" ])
771
+ seasonally_labeled_daily_data [daily_rain .name ]
772
+ .groupby (seasonally_labeled_daily_data ["seasons_starts" ])
773
773
.map (
774
774
onset_date ,
775
775
wet_thresh = wet_thresh ,
@@ -784,7 +784,7 @@ def seasonal_onset_date(
784
784
.rename ({"seasons_starts" : time_dim })
785
785
)
786
786
# Get the seasons ends
787
- seasons_ends = grouped_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
787
+ seasons_ends = seasonally_labeled_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
788
788
seasonal_onset_date = xr .merge ([seasonal_data , seasons_ends ])
789
789
790
790
# Tip to get dates from timedelta search_start_day
@@ -852,21 +852,21 @@ def seasonal_cess_date_from_sm(
852
852
end_month = first_end_date .dt .month .values
853
853
854
854
# Apply daily grouping by season
855
- grouped_daily_data = daily_tobegroupedby_season (
855
+ seasonally_labeled_daily_data = daily_tobegroupedby_season (
856
856
soil_moisture , search_start_day , search_start_month , end_day , end_month
857
857
)
858
858
# Apply cess_date
859
859
seasonal_data = (
860
- grouped_daily_data [soil_moisture .name ]
861
- .groupby (grouped_daily_data ["seasons_starts" ])
860
+ seasonally_labeled_daily_data [soil_moisture .name ]
861
+ .groupby (seasonally_labeled_daily_data ["seasons_starts" ])
862
862
.map (
863
863
cess_date_from_sm ,
864
864
dry_thresh = dry_thresh ,
865
865
dry_spell_length_thresh = dry_spell_length_thresh ,
866
866
)
867
867
).rename ("cess_delta" )
868
868
# Get the seasons ends
869
- seasons_ends = grouped_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
869
+ seasons_ends = seasonally_labeled_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
870
870
seasonal_cess_date = xr .merge ([seasonal_data , seasons_ends ])
871
871
872
872
# Tip to get dates from timedelta search_start_day
@@ -905,13 +905,13 @@ def seasonal_cess_date_from_rain(
905
905
end_month = first_end_date .dt .month .values
906
906
907
907
# Apply daily grouping by season
908
- grouped_daily_data = daily_tobegroupedby_season (
908
+ seasonally_labeled_daily_data = daily_tobegroupedby_season (
909
909
daily_rain , search_start_day , search_start_month , end_day , end_month
910
910
)
911
911
# Apply cess_date
912
912
seasonal_data = (
913
- grouped_daily_data [daily_rain .name ]
914
- .groupby (grouped_daily_data ["seasons_starts" ])
913
+ seasonally_labeled_daily_data [daily_rain .name ]
914
+ .groupby (seasonally_labeled_daily_data ["seasons_starts" ])
915
915
.map (
916
916
cess_date_from_rain ,
917
917
dry_thresh = dry_thresh ,
@@ -922,7 +922,7 @@ def seasonal_cess_date_from_rain(
922
922
)
923
923
).rename ("cess_delta" )
924
924
# Get the seasons ends
925
- seasons_ends = grouped_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
925
+ seasons_ends = seasonally_labeled_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
926
926
seasonal_cess_date = xr .merge ([seasonal_data , seasons_ends ])
927
927
928
928
# Tip to get dates from timedelta search_start_day
@@ -973,16 +973,16 @@ def seasonal_sum(
973
973
Examples
974
974
--------
975
975
"""
976
- grouped_daily_data = daily_tobegroupedby_season (
976
+ seasonally_labeled_daily_data = daily_tobegroupedby_season (
977
977
daily_data , start_day , start_month , end_day , end_month
978
978
)
979
979
seasonal_data = (
980
- grouped_daily_data [daily_data .name ]
981
- .groupby (grouped_daily_data ["seasons_starts" ])
980
+ seasonally_labeled_daily_data [daily_data .name ]
981
+ .groupby (seasonally_labeled_daily_data ["seasons_starts" ])
982
982
.sum (dim = time_dim , skipna = True , min_count = min_count )
983
983
# .rename({"seasons_starts": time_dim})
984
984
)
985
- seasons_ends = grouped_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
985
+ seasons_ends = seasonally_labeled_daily_data ["seasons_ends" ].rename ({"group" : time_dim })
986
986
summed_seasons = xr .merge ([seasonal_data , seasons_ends ])
987
987
return summed_seasons
988
988
0 commit comments