有没有办法将numpy svd的输出显示为对角矩阵？

Question

numpy提供了linalg.svd来计算奇异值分解。

我写来帮助理解svd的小例子

A = np.array([[1, 0, 0, 0],
       [0, 0, 0, 4],
       [0, 3, 0, 0],
       [0, 0, 0, 0],
       [2, 0, 0, 0]])
U, S, VT = svd(A)
print(S)

输出

[ 4.          3.          2.23606798 -0.        ]

以数学形式显示，输出看起来像（fig_1）

是否可以将numpy.linalg.svd的输出显示为fig_1？