def get_change_table_for_id_scaling(base_grid, resource):
n_plant = param["n_plant_to_scale"]
plants = get_plant_with_resource(base_grid, resource)
ct = ChangeTable(base_grid)
ct.scale_plant_capacity(
resource,
plant_id={
z: f
for z, f in zip(
np.random.choice(plants, size=n_plant, replace=False),
2 * np.random.random(size=n_plant),
)
},
)
return ct.ct
def get_change_table_for_zone_scaling(base_grid, resource):
n_zone = param["n_zone_to_scale"]
zones = get_zone_with_resource(base_grid, resource)
ct = ChangeTable(base_grid)
ct.scale_plant_capacity(
resource,
zone_name={
z: f
for z, f in zip(
np.random.choice(zones, size=n_zone, replace=False),
2 * np.random.random(size=n_zone),
)
},
)
return ct.ct
