@@ -261,30 +261,32 @@ def crop_suitability(
261
261
seasonal_tmax = tmax .sel (T = tmax ['T.season' ]== target_season )
262
262
seasonal_tmin = tmin .sel (T = tmin ['T.season' ]== target_season )
263
263
264
- seasonal_avg_tmax_suitability = 1 * (
264
+ seasonal_avg_tmax_suitability = (
265
265
seasonal_tmax .groupby ("T.year" ).mean () <= max_temp
266
266
)
267
- seasonal_avg_tmin_suitability = 1 * (
267
+ seasonal_avg_tmin_suitability = (
268
268
seasonal_tmin .groupby ("T.year" ).mean () >= min_temp
269
269
)
270
270
271
- seasonal_avg_temp_amplitude_suitability = 1 * (
271
+ seasonal_avg_temp_amplitude_suitability = (
272
272
(seasonal_tmax - seasonal_tmin ).groupby ("T.year" ).mean () <= temp_range
273
273
)
274
274
275
- seasonal_wet_days_suitability = 1 * (
275
+ seasonal_wet_days_suitability = (
276
276
(seasonal_precip >= wet_day_def ).groupby ("T.year" ).sum () >= min_wet_days
277
277
)
278
278
279
- seasonal_total_precip_suitability = 1 * (
279
+ seasonal_total_precip_suitability = (
280
280
(seasonal_precip .groupby ("T.year" ).sum () <= upper_wet_threshold ) &
281
281
(seasonal_precip .groupby ("T.year" ).sum () >= lower_wet_threshold )
282
282
)
283
283
284
284
crop_suit = (
285
- seasonal_avg_tmax_suitability + seasonal_avg_tmin_suitability +
286
- seasonal_avg_temp_amplitude_suitability + seasonal_total_precip_suitability +
287
- seasonal_wet_days_suitability
285
+ seasonal_avg_tmax_suitability .astype (int ) +
286
+ seasonal_avg_tmin_suitability .astype (int ) +
287
+ seasonal_avg_temp_amplitude_suitability .astype (int ) +
288
+ seasonal_total_precip_suitability .astype (int ) +
289
+ seasonal_wet_days_suitability .astype (int )
288
290
)
289
291
290
292
crop_suitability = xr .Dataset (
0 commit comments