numpy.diagflat()

numpy.diagflat

numpy.diagflat(v, k=0) [source]

Create a two-dimensional array with the flattened input as a diagonal.

登录查看完整内容
Parameters:

v : array_like

Input data, which is flattened and set as the k-th diagonal of the output.

k : int, optional

Diagonal to set; 0, the default, corresponds to the “main” diagonal, a positive (negative) k giving the number of the diagonal above (below) the main.