1. 程式人生 > 程式設計 >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) #輸出最終結果

結果為: 這裡寫圖片描述

其結果即為我們求解的,需要兩步才可完成。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。