numpy 矩陣操作
阿新 • • 發佈：2018-10-10
print port color index import 對角線 操作 pan == numpy 對矩陣對角線、上三角、下三角以及它們所在位置索引的提取
import numpy as np a = np.random.randint(0,10,[5,5]) print(a) # c = np.triu(a,0) #上三角 # print(c) # d = np.tril(a,0) # 下三角 # print(d) # 尋找上三角形的位置 up = np.triu(a,1) up_bool = a==up up_palce = np.argwhere(up_bool==True) for i,j in up_palce: if i<=j: #下三角就是x大於等於y，他兩想相反。 a[i,j] = 66 print(a) # 尋找對角線的位置 c = np.diag(a,0) print(c) s = a==c index = np.argwhere(s==True) print(index) for x,y in index: if x==y: a[x,y] =0 print(a)
numpy 矩陣操作