numpy中三維數組轉變成二維數組


numpy中reshape()函數對三維數組進行轉換成二維數組,見下面例子:

 1 >>>a=np.reshape(np.arange(18),(3,3,2))
 2 >>> a
 3 array([[[ 0,  1],
 4         [ 2,  3],
 5         [ 4,  5]],
 6 
 7        [[ 6,  7],
 8         [ 8,  9],
 9         [10, 11]],
10 
11        [[12, 13],
12         [14, 15],
13         [16, 17]]])
14 >>>a=reshape(a,(-1,3))
15 >>>a
16 array([[ 0,  1,  2],
17        [ 3,  4,  5],
18        [ 6,  7,  8],
19        [ 9, 10, 11],
20        [12, 13, 14],
21        [15, 16, 17]])
22 
23 >>>a=np.reshape(np.arange(18),(2,3,3))
24 >>>a
25 array([[[ 0,  1,  2],
26         [ 3,  4,  5],
27         [ 6,  7,  8]],
28 
29        [[ 9, 10, 11],
30         [12, 13, 14],
31         [15, 16, 17]]])
32 >>>a=np.reshape(a,(-1,3))
33 >>>array([[ 0,  1,  2],
34        [ 3,  4,  5],
35        [ 6,  7,  8],
36        [ 9, 10, 11],
37        [12, 13, 14],
38        [15, 16, 17]])

易見,對reshape()里面的數組形狀第一個為-1,第二個為第二維元素的數目。那么就會對原來的所有元素進行一個平均分配得到一個二維數組。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM