python矩阵对角线求和numpy_以numpy计算对角线和（更快）

使用stride_tricks有一个可能的解决方案。这在一定程度上是基于this question的答案中提供的大量信息，但我认为，问题只是不同而已，不算重复。这是一个基本思想，应用到一个方阵中，见下面的函数实现更一般的解决方案。在>>> cols = 8

>>> a = numpy.arange(cols * cols).reshape((cols, cols))

>>> fill = numpy.zeros((cols - 1) * cols, dtype='i8').reshape((cols - 1, cols))

>>> stacked = numpy.vstack((a, fill, a))

>>> major_stride, minor_stride = stacked.strides

>>> strides = major_stride, minor_stride * (cols + 1)

>>> shape = (cols * 2 - 1, cols)

>>> numpy.lib.stride_tricks.as_strided(stacked, shape, strides)

array([[ 0, 9, 18, 27, 36, 45, 54, 63],

[ 8, 17, 26, 35, 44, 53, 62, 0],

[16, 25, 34, 43, 52, 61, 0, 0],

[24, 33, 42, 51, 60, 0, 0, 0],

[32, 41, 50, 59, 0, 0, 0, 0],

[40, 49, 58, 0, 0, 0, 0,