Calculating energy production from weather forecast in Python