在 numpy 中添加一列

在使用 numpy 进行数据处理时，我们有时需要添加一列数据。本文将介绍如何使用 numpy 在数组中添加一列数据。

示例代码

import numpy as np

# 创建一个 4 行 3 列的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])

# 创建一个与 arr 行数相同的一维数组
new_col = np.array([13, 14, 15, 16])

# 使用 np.concatenate() 函数在 arr 最后一列添加 new_col
arr_with_new_col = np.concatenate((arr, new_col.reshape(-1, 1)), axis=1)

print(arr_with_new_col)

输出结果为：

array([[ 1,  2,  3, 13],
       [ 4,  5,  6, 14],
       [ 7,  8,  9, 15],
       [10, 11, 12, 16]])

代码解析

1. 导入 numpy 库。

import numpy as np

2. 创建一个二维数组。

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])

3. 创建一个与 arr 行数相同的一维数组。

new_col = np.array([13, 14, 15, 16])

4. 使用 np.concatenate() 函数在 arr 最后一列添加 new_col。

arr_with_new_col = np.concatenate((arr, new_col.reshape(-1, 1)), axis=1)

• np.concatenate() 函数用于连接两个或多个数组。
• axis 参数指定连接的轴，axis=1 表示连接的轴为列。

5. 输出结果。

print(arr_with_new_col)

结论

使用 numpy 中的 np.concatenate() 函数，可以在数组中添加一列数据，从而方便地完成数据处理。