Sum of values based on list of values in python