擅长:python、mysql、java
<p>可以使用<a href="https://numpy.org/doc/stable/reference/generated/numpy.eye.html" rel="nofollow noreferrer">^{<cd1>}</a>获得对角线的遮罩(如果反转遮罩,则会得到非对角线)</p>
<pre><code>>> a = np.linspace(1,9,9).reshape((3,3))
array([[1., 2., 3.],
[4., 5., 6.],
[7., 8., 9.]])
>> a[~np.eye(*a.shape,dtype=bool)]
array([2., 3., 4., 6., 7., 8.])
</code></pre>