将数组求和到矩阵对角线（元素和）

Question

我需要求一个矩阵对角线一个数组（元素明智的和）：

import numpy as np



mx = np.matrix([[0,0,0,0],[1,1,1,1],[2,3,4,5],[6,7,8,9]])
print mx
print '\n'

v = np.array([20,0,10,0.5])
print v
print '\n'

这不起作用：

mx = np.diagonal(mx,v)
print mx

我该怎么办？谢谢大家