可以使用 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 连接起来,即沿着列的方向连接。

Python NumPy 二维数组添加列:使用 concatenate 函数

原文地址: https://www.cveoy.top/t/topic/mVME 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录