1. 程式人生 > 程式設計 >Python reshape的用法及多個二維數組合併為三維陣列的例項

Python reshape的用法及多個二維數組合併為三維陣列的例項

reshape(shape) : 不改變陣列元素,返回一個shape形狀的陣列,原陣列不變。是對每行元素進行處理

resize(shape) : 與.reshape()功能一致,但修改原陣列

In [1]: a = np.arange(20)
#原陣列不變
In [2]: a.reshape([4,5])
Out[2]:
array([[ 0,1,2,3,4],[ 5,6,7,8,9],[10,11,12,13,14],[15,16,17,18,19]])
 
In [3]: a
Out[3]:
array([ 0,4,5,9,10,14,15,19])
 
#修改原陣列
In [4]: a.resize([4,5])
 
In [5]: a
Out[5]:
array([[ 0,19]])

.swapaxes(ax1,ax2) : 將陣列n個維度中兩個維度進行調換,不改變原陣列

In [6]: a.swapaxes(1,0)
Out[6]:
array([[ 0,15],[ 1,16],[ 2,17],[ 3,18],[ 4,19]])

.flatten() : 對陣列進行降維,返回摺疊後的一維陣列,原陣列不變

In [7]: a.flatten()
Out[7]:
array([ 0,19])

將多個二維數組合併為一個三維陣列

方法一:

對於兩個(或者多個)同一維度的矩陣,直接利用np.array()重新構造一個array,這樣可以變相起到擴充套件維數的作用。例如:

import numpy as np
 
a = np.array([[1,3],[4,6]])
b = np.array([[2,6]])
c = np.array([[3,6]])
print('矩陣a:\n',a)
print('維數:',a.shape)
 
com = np.array([a,b,c])
print('合併矩陣:\n',com)
print('維數:',com.shape)
 
輸出結果為:
 
矩陣a:
 [[1 2 3]
 [4 5 6]]
維數: (2,3)
合併矩陣:
 [[[1 2 3]
 [4 5 6]]
 
 [[2 2 3]
 [4 5 6]]
 
 [[3 2 3]
 [4 5 6]]]
維數: (3,3)

方法二:

但是,如果兩個array,使用方法一時會出現如下結果:

import numpy as np
 
aa = np.array([[[1,6]],[[2,[[3,6]]])
a = np.array([[4,6]])
 
com = np.array([aa,a])
print('合併矩陣:\n',com.shape)
 
輸出結果:
 
合併矩陣:
 [array([[[1,6]]])
 array([[4,6]])]
維數: (2,)

可以看到:輸出的維數不對,以上方法就不適用了。

那麼,我們就需要利用np.append和array.reshape()函式對陣列進行拼接之後重組,具體實現如下:

import numpy as np
 
aa = np.array([[[1,6]])
data = np.append(aa,a)#先拼接成一個行向量
print(data)
 
dim = aa.shape#獲取原矩陣的維數
print('原矩陣維數:',dim)
data1 = data.reshape(dim[0]+1,dim[1],dim[2])#再通過原矩陣的維數重新組合
 
print('合併矩陣:\n',data1)
print('維數:',data1.shape)

現在來看一下用reshape將二維資料升為三維後的資料分佈情況:

import numpy as np
b = np.arange(36).reshape((6,6))
b1 = b.reshape(2,6)

b的元素:

Python reshape的用法及多個二維數組合併為三維陣列的例項

b1的元素:

Python reshape的用法及多個二維數組合併為三維陣列的例項

可以看到,原來6*6的矩陣被分為了2個3*6的矩陣。每一行的資料分佈並沒有改變,只是將前3行劃為一個維度,然後將後三行劃為另一個維度。

b1.reshape(6,6)

如果用這條命令,則資料又被還原了回去,與b的一樣。

b1.reshape(3,12)

Python reshape的用法及多個二維數組合併為三維陣列的例項

如果用的是reshape(3,12),則相當於將資料首先拉伸為1維的,然後再將一維資料重組為3*12

方法三:

相比於前兩種方法,這種方法可謂“曲線救國”之典範,具體思路是:先轉化成list,拼接後再轉化回去。

這是因為list中的append()函式可以在新增函式的時候不改變原來list的維度。雖然沒有對這種方法進行一個速度測試,但直覺來看時間複雜度挺高的,建議慎用。

aa = np.array([[[1,6]])
 
#將array轉換成list
aa = aa.tolist(aa)
a = a.list(a)
 
aa.append(a)#注意與方法二中np.append()用法的區別
com = np.array(aa)
print(com.shape)
 
輸出結果:
 
合併矩陣:
  [[[1 2 3]
  [4 5 6]]
 
  [[2 2 3]
  [4 5 6]]
 
  [[3 2 3]
  [4 5 6]]
  
  [[4 2 3]
  [4,6]]]
維數: (4,3)

這裡注意:

兩種型別的相互轉換函式:

array轉list:a = a.tolist()

list轉array:a =np.array(a)

以上這篇Python reshape的用法及多個二維數組合併為三維陣列的例項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。