how to solve the groundwater remediation problem

remediation.py

from pflotran import Flow, Transport
from randomsoils import soil_generator
import pandas as pd

nYears = 30; time_step = 1/12 # allow change every month
grid, wells = pd.read_csv('basic_grid_and_wells.csv')

for ix in range(nRuns):   # run a few simulations
  modified_soil = soil_generator(grid)
  pumping_schedule = pd.random(wells, range(0,nYears,time_step))
  contamination_levels = Transport(Flow(pumping_schedule, grid))
  
  pumping_cost = well_cost*pumping_schedule
  penalty_cost = sum(max(contamination_levels-THRESHOLD,0), observation_wells)
  print('TOTAL_COST = %f', pumping_cost + penalty_cost)

see about PEST: http://www.pesthomepage.org/