擅长:python、mysql、java
<p>您可能正在寻找<code>numpy.trace()</code>(文档中的<a href="https://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.trace.html" rel="noreferrer">here</a>)来直接获得跟踪,或者{<cd2>}来获得对角线向量<a href="https://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.diagonal.html" rel="noreferrer">documented here</a></p>
<p>首先,使用<code>rectdf.as_matrix()</code>将数据帧转换为numpy矩阵</p>
<p>然后:</p>
<pre><code>np.trace(matrix, offset)
</code></pre>
<p>偏移量可以是正的也可以是负的,可以满足您所需的偏移量。在</p>
<p>例如,如果我们这样做:</p>
^{pr2}$
<p>我们得到输出:</p>
<pre><code>12
22
30
21
12
6
2
</code></pre>
<p>对于一般矩阵,我们需要从<code>-(rows - 1)</code>到{<cd5>},也就是说,如果我们有一个变量<code>rows</code>和一个变量<code>columns</code>:</p>
<pre><code>a = np.arange(rows * columns).reshape(rows, columns)
for x in range(-(rows - 1), columns): print np.trace(a, x)
</code></pre>