擅长:python、mysql、java
<p>使用np.diag(<a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.diag.html" rel="nofollow noreferrer">https://docs.scipy.org/doc/numpy/reference/generated/numpy.diag.html</a>)</p>
<pre><code>>>> np.diag([0, 4, 8])
array([[0, 0, 0],
[0, 4, 0],
[0, 0, 8]])
</code></pre>