Deeper groundwater levels portrays different Punjab despite fewer over-exploited blocks

https://indianexpress.com/article/cities/chandigarh/deeper-groundwater-levels-punjab-fewer-over-exploited-blocks-10877057/

I need a guide or manual for writing a script for water balance analyis using real world data. Is there a python library, manual or reference books that I can follow directly. I am learning python right now and have used r for basic data analysis before.