numpy.diagflat#
- numpy.diagflat(v, k=0)[源代码]#
创建一个二维数组,将展平的输入作为对角线.
- 参数:
- varray_like
输入数据,该数据被展平并设置为输出的第 k 个对角线.
- k整数,可选
要设置的对角线; 0(默认值)对应于“主”对角线,正(负) k 给出主对角线上方(下方)的对角线数.
- 返回:
- outndarray
二维输出数组.
示例
>>> import numpy as np >>> 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]])