python數組添加整行或整列

代碼如下,見註釋:

import numpy as np
m_list=[[1,2,3],[3,4,5]]
m_arr=np.array(m_list) # 轉爲數組
print(np.append(m_arr,[[1,1,1]],axis=0)) # 添加整行元素
print(np.append(m_arr,[[1],[1]],axis=1)) # 添加整列
ones_row=np.ones(len(m_arr[0]),int) # 生成全爲1的一維列表,即【1,1,1】
print(np.append(m_arr,[ones_row],axis=0)) 
ones_col=[[1] for i in range(len(m_arr))] # 生成全爲1的二維嵌套列表,即[[1],[1]]
print(np.append(m_arr,ones_col,axis=1))

結果如下:

[[1 2 3]
 [3 4 5]
 [1 1 1]]
 
[[1 2 3 1]
 [3 4 5 1]]
 
[[1 2 3]
 [3 4 5]
 [1 1 1]]
 
[[1 2 3 1]
 [3 4 5 1]]
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章