自己寫的一些函式--對角陣為1,小對角陣全是1

weixin_42053726發表於2018-12-30

產生一個矩陣如下圖

'''
對角陣為1,小對角陣全是1
'''
import numpy as np #匯入必要的包
def big_matrix(small = 3, big = 3 , value = 1):
    big_M = []
    for i in np.arange(0,small*big,small):
        for j in np.arange(small):
            a = np.zeros(small*big)
            a[i:i+small] = value
            big_M.append(a)

    big_M = np.array(big_M)
    return big_M

big_matrix()

相關文章