python reshape()二維矩陣變一維

代碼

import numpy as np

a=np.arange(8).reshape(2,4)
print(a)
print(a.shape)

b=a.reshape(8)
print(b)
print(b.shape)

運行結果

[[0 1 2 3]
[4 5 6 7]]
(2, 4)
[0 1 2 3 4 5 6 7]
(8,)