擅长:python、mysql、java
<p>你不需要任何索引操作。只需使用<code>tril</code>、<code>triu</code>和<code>identity</code>函数:</p>
<pre><code>import numpy as np
A = np.array([[ 8. , -6. , 2. ],
[-0.5, 8. , -6. ],
[ 0.5, -0.5, 2. ]])
U = np.triu(A)
#[[ 8. -6. 2.]
# [-0. 8. -6.]
# [ 0. -0. 2.]]
L = np.tril(A, k=-1) + np.identity(3)
#[[ 1. 0. 0. ]
# [-0.5 1. 0. ]
# [ 0.5 -0.5 1. ]]
</code></pre>