突然用到 副对角线 全为1的矩阵, 不知道怎么调用numpy, 自己写了一个:
import numpy as np
# 副对角线 全为 1
def get_subdiag(n, value):
ar = []
for i in range(n):
a = []
for j in range(n):
if i + j == n - 1:
a.append(value)
else:
a.append(0)
ar.append(a)
return np.array(ar)
ar = get_subdiag(3, 1)
print(ar)
输入:
[[0 0 1]
[0 1 0]
[1 0 0]]