numpy记录（二）—— 提取矩阵行或列、转置

numpy记录（二）—— 提取矩阵行或列、转置

Date: June 7, 2022
author: jinxili

提取矩阵某一行 或 某一列

a=np.arange(9).reshape(3,3)

a:
[[0 1 2] 
 [3 4 5] 
 [6 7 8]]

取任意一行

a[0]:
[0 1 2]

取任意一列

a[:,2]:
[2 5 8]

转置

ndarray的T属性，在数组后跟.T即可
.T，不指定转置规则的默认规则

arr1 = np.arange(6)
print("arr1: \n", arr1)
print("arr1.T:\n", arr1.T)

arr2 = np.arange(8).reshape(2,4) 
print("arr2: \n", arr2)
print("arr2.T: \n", arr2.T)

arr3 = np.arange(24).reshape((2,3,4)) 
print("arr3: \n", arr3)
print("arr3.T: \n", arr3.T)
print("np.shape(arr3): \n", np.shape(arr3))
print("np.shape(arr3.T): \n", np.shape(arr3.T))

arr1:
 [0 1 2 3 4 5]
arr1.T:
 [0 1 2 3 4 5]

arr2:
 [[0 1 2 3]
 [4 5 6 7]]
arr2.T:
 [[0 4]
 [1 5]
 [2 6]
 [3 7]]

arr3:
 [[[ 0  1  2  3]
  [ 4  5  6  7]
  [ 8  9 10 11]]

 [[12 13 14 15]
  [16 17 18 19]
  [20 21 22 23]]]
arr3.T:
 [[[ 0 12]
  [ 4 16]
  [ 8 20]]

 [[ 1 13]
  [ 5 17]
  [ 9 21]]

 [[ 2 14]
  [ 6 18]
  [10 22]]

 [[ 3 15]
  [ 7 19]
  [11 23]]]
 
np.shape(arr3):
 (2, 3, 4)

np.shape(arr3.T):
 (4, 3, 2)