1. 程式人生 > >Numpy 矩陣和陣列拼接

Numpy 矩陣和陣列拼接

寫程式碼需要將一個數組和矩陣拼接在一起,查了下numpy的函式,發現有下面幾種函式,其中所有函式都需要保持元素的維度一致,因此不能直接拼接,需要將陣列升維,使得陣列成為一個“矩陣”才可以進行拼接,以append函式為例做個示範

  • np.append()
  • np.concatenate()
  • np.stack()
  • np.hstack()
  • np.vstack()
  • np.dstack()
import numpy as np
a = np.array([1, 1, 1])
a = np.expand_dims(a, axis=1)
b = np.array([[2,2,2],[2,2,2],[2,2,2]])
print(a)
print(b)
c = np.append(a,b ,axis=1)
print(c)

結果如圖所示: