リストの内包表記かな。
あと、print文も
print "sum[%d] =%d" % (r, sum[r])
みたいにするのがいいと思う。