numpy.diagflat()

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

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

Parameters:

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.
Returns:	out : ndarray The 2-D output array.

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.

Returns:

out : ndarray

The 2-D output array.

Examples

>>> np.diagflat([[1,2], [3,4]])
array([[1, 0, 0, 0],
       [0, 2, 0, 0],
       [0, 0, 3, 0],
       [0, 0, 0, 4]])

>>> np.diagflat([1,2], 1)
array([[0, 1, 0],
       [0, 0, 2],
       [0, 0, 0]])

Links:

https://docs.scipy.org/doc/numpy-1.11.0/reference/generated/numpy.diagflat.html

doc_NumPy

2017-01-10 18:13:38

Comments