二維 NumPy 陣列的子集提取
如果你的二維 numpy 陣列結構規整,即每一列和每一欄都包含固定數量的數值,複雜的子集提取操作就會變得非常簡單。請看下面的程式碼,其中元素 "a" 和 "c" 是從一個巢狀串列中提取出來的。
# numpy
import numpy as np
np_x = np.array(x)
np_x[:, 0]
逗號前的索引代表「列」,逗號後的索引代表「欄」。冒號 : 表示切片;在此範例中,它的意思是讓 Python 選擇所有列。
本練習屬於課程
Python 入門
練習說明
- 印出
np_baseball的第 50 列。 - 建立一個新變數
np_weight_lb,用來提取np_baseball的整個第二欄。 - 選擇
np_baseball中第 124 位棒球球員的身高(第一欄)並將其印出。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
import numpy as np
np_baseball = np.array(baseball)
# Print out the 50th row of np_baseball
# Select the entire second column of np_baseball: np_weight_lb
# Print out height of 124th player