1. 程式人生 > >Numpy 改變陣列維度的幾種方法

Numpy 改變陣列維度的幾種方法

來自 《Python資料分析基礎教程:Numpy 學習指南(第2版)》

Numpy改變陣列維度的方法有:

  • reshape()
  • ravel()
  • flatten()
  • 用元組設定維度
  • transpose()
  • resize()

下面將依次進行說明

    0. 首先,建立一個多維陣列

from numpy import * 
a = arange(24)

    得到:

[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]

    1.reshape 函式

b = a.reshape(2,3,4)
print(b)

    得到一個 2*3*4 維的陣列:

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

 [[12 13 14 15]
  [16 17 18 19]
  [20 21 22 23]]]

    可以想象成2層小洋樓,每層有 3*4 個房間

    2.ravel函式 可以將多維陣列展平(也就是變回一維)

c = b.ravel()
print(c)

    得到一維陣列

[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]

    3.flatten函式  也是將多維陣列展平,與ravel函式的功能相同,不過flatten函式會請求分配記憶體來儲存結果,而ravel函式只是返回陣列的一個檢視(view)

c = b.flatten()
print(c)

    得到一維陣列

[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]

    4.用元組設定維度

    直接用一個正整數元組來設定陣列的維度

b.shape = (6,4)
print(b)
   這種做法將直接改變所操作的陣列,現在陣列b變成了一個 6*4 的多維陣列
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]
 [16 17 18 19]
 [20 21 22 23]]

    5.transpose函式 將矩陣進行轉置

d = b.transpose()
print(d)
    得到 4*6 的多維陣列
[[ 0  4  8 12 16 20]
 [ 1  5  9 13 17 21]
 [ 2  6 10 14 18 22]
 [ 3  7 11 15 19 23]]

    6.resize函式  和reshape函式的功能一樣,但resize會直接修改所操作的陣列

b.resize((2,12))
print(b)

    得到 2*12 的兩維陣列

[[ 0  1  2  3  4  5  6  7  8  9 10 11]
 [12 13 14 15 16 17 18 19 20 21 22 23]]

    並且這一步不可以通過賦值來實現,如下所示:

e = b.resize((2,12))
print(e)

    此時結果為:

None