给numpy数组添加行或列的几种方法

d = np.array([[1, 2], [2, 3], [3, 4]])
d
array([[1, 2],
       [2, 3],
       [3, 4]])

现在要添加一列[5, 6, 7]，或者一行[5, 6]

np.c_、np.r_

# 添加列
np.c_[d, [5, 6, 7]]
array([[1, 2, 5],
       [2, 3, 6],
       [3, 4, 7]])

# 添加行，注意维度
np.r_[d, [[5, 6]]]
array([[1, 2],
       [2, 3],
       [3, 4],
       [5, 6]])

np.append()

# 添加列，注意所添加数组格式
np.append(d, [[5], [6], [7]], axis=1)
array([[1, 2, 5],
       [2, 3, 6],
       [3, 4, 7]])

# 添加行
np.append(d, [[5, 6]], axis=0)
array([[1, 2],
       [2, 3],
       [3, 4],
       [5, 6]])

# 注意一定要加axis，不然就变成一维的了
np.append(d, [[5, 6]])
array([1, 2, 2, 3, 3, 4, 5, 6])

np.insert()

# 第二个参数为插入的索引位置
# axis必须写，不然得到一维数组
# 注意：当插入位置为-1时，实际插入位置不是最后一个位置，而是倒数第二个
# 插入列
np.insert(d, 2, [5, 6, 7], axis=1)
array([[1, 2, 5],
       [2, 3, 6],
       [3, 4, 7]])

# 插入行，位置为-1，实际插入在倒数第二个位置
np.insert(d, -1, [5, 6], axis=0)
array([[1, 2],
       [2, 3],
       [5, 6],
       [3, 4]])

# 插入到最后的位置
np.insert(d, d.shape[0], [5, 6], axis=0)
array([[1, 2],
       [2, 3],
       [3, 4],
       [5, 6]])

np.column_stack()、np.row_stack()

np.column_stack((d, [5, 6, 7]))
array([[1, 2, 5],
       [2, 3, 6],
       [3, 4, 7]])

np.row_stack((d, [5, 6]))
array([[1, 2],
       [2, 3],
       [3, 4],
       [5, 6]])

np.hstack((d, [[5], [6], [7]]))
array([[1, 2, 5],
       [2, 3, 6],
       [3, 4, 7]])

np.vstack((d, [5, 6]))
array([[1, 2],
       [2, 3],
       [3, 4],
       [5, 6]])