python numpy 二维数组 增加一列
可以使用numpy的concatenate函数来增加一列。
例如,假设有一个二维数组arr,要在其右侧增加一列,可以使用以下代码:
import numpy as np
# 创建二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 创建要添加的一列数据
new_col = np.array([10, 11, 12])
# 使用concatenate函数将新列添加到数组的右侧
new_arr = np.concatenate((arr, new_col.reshape(-1, 1)), axis=1)
# 打印新数组
print(new_arr)
输出:
array([[ 1, 2, 3, 10],
[ 4, 5, 6, 11],
[ 7, 8, 9, 12]])
其中,concatenate函数的第一个参数是一个元组,包含要连接的数组,第二个参数是指定连接的轴。
在本例中,将新列reshape为一个列向量,即(-1, 1),并将其与原数组使用axis=1连接起来,即沿着列的方向连接。
原文地址: https://www.cveoy.top/t/topic/bsqQ 著作权归作者所有。请勿转载和采集!