On 2/3/2012 3:37 PM, josef.p...@gmail.com wrote: > res = - np.dot(x.T, mass*x) > res[np.arange(3), np.arange(3)] -= np.trace(res)
Nice! Get some speed gain with slicing: res = - np.dot(x.T, mass*x) res.flat[slice(0,None,4)] -= np.trace(res) Alan _______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@scipy.org http://mail.scipy.org/mailman/listinfo/numpy-discussion