1. 程式人生 > >Python取numpy陣列的某幾行某幾列方法

Python取numpy陣列的某幾行某幾列方法

Python 取numpy陣列的某幾行某幾列方法

直接分析,如原矩陣如下(1):
這裡寫圖片描述 (1)
我們要擷取的矩陣(取其一三行,和三四列資料構成矩陣)為如下(2):
這裡寫圖片描述 (2)

錯誤分析:

取 C 的1 3行,3 4 列,定義

Z = [0,2]   #定義行數
d = [2,3]   #定義列數

#程式碼
C_zd = C[z,d] 

則結果為:這裡寫圖片描述
由結果分析取的是第一行第三列和第三行第四列的資料,並非我們想要的結果。

正確分析:

C_A = c[[0,2]]
#先取出想要的行資料 C_A = C_A[:,[2,3]] #再取出要求的列資料 print(C_A) #輸出最終結果

結果為: 這裡寫圖片描述
其結果即為我們求解的,需要兩步才可完成。