Matrix Elements Sum Python