Python NumPy diagflat方法用法及代码示例

Numpy 的 diagflat(~) 方法创建一个 2D Numpy 数组，其对角线由展平的输入数组指定。所有其他条目均用零填充。

1. a | array-like

输入数组。多维数组将自动展平为一维。

2. k | int | optional

如果 k 为正，则顶部的对角线将被填充。如果为负，则底部的对角线将被填充。默认情况下，k=0 。

二维 Numpy 数组。

考虑以下二维数组：

np.diagflat([4,5,6])



array([[4, 0, 0],
       [0, 5, 0],
       [0, 0, 6]])

np.diagflat([4,5,6], k=1)



array([[0, 4, 0, 0],
       [0, 0, 5, 0],
       [0, 0, 0, 6],
       [0, 0, 0, 0]])

np.diagflat([4,5], k=1)



array([[0, 4, 0],
       [0, 0, 5],
       [0, 0, 0]])

np.diagflat([4,5], k=-1)



array([[0, 0, 0],
       [4, 0, 0],
       [0, 5, 0]])

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | diagflat method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。