Numpy 改變陣列維度的幾種方法
阿新 • • 發佈:2019-02-10
來自 《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