更改跟踪行为 Numpy

Question

我有这个 5x5 眼睛阵列，我应该使用 'trace' 关键字从 k=1 计算总和，但结果我必须更改跟踪起始位置，我该怎么做？。

import numpy as np
# TODO: Trace an eye  

eye_0 = np.eye(5,5, k=1)
trace_eye = np.trace(eye_0)
print(eye_0)
trace_eye 

[[0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 1.]
 [0. 0. 0. 0. 0.]]
0.0

Answer 1

Numpy.trace 有一个偏移量参数：

np.trace(eye_0, offset = 1)

这将在您的情况下返回 4。