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

取numpy陣列的某幾行某幾列

這個操作在numpy陣列上的操作感覺有點麻煩,但是也沒辦法。  例如  a = [[1,2,3],  [4,5,6],  [7,8,9]]  取 a 的 2 3 行, 1 2 列

c=[1,2]  d =[0,1]

若寫為  b = a[c,d]  output:  [4 8]  取的是 第二行第一列 和第三行第二列的資料  這並不是我們想要的結果。

正確做法是:  b = a[c]先取想要的行資料  b = b[:,d]  print(b)  output:  [[4 5]  [7 8]]  這才是我們想要的結果。必須要經過這兩步才能完成 ---------------------  作者:dream_toy  來源:CSDN  原文:https://blog.csdn.net/u010882121/article/details/76045303  版權宣告:本文為博主原創文章,轉載請附上博文連結!