如何使用Numpy来向数组中添加列|极客教程

如何使用Numpy来向数组中添加列

在数据处理和科学计算中，经常需要对数组进行操作，比如添加列。Numpy是Python中一个非常重要的库，它提供了大量的数组操作功能。本文将详细介绍如何使用Numpy来向数组中添加列。

1. 基本概念

在Numpy中，数组可以是一维的，也可以是多维的。向数组添加列通常是指在二维数组中添加一列。这可以通过numpy.append()函数实现，该函数可以沿指定轴向数组添加值。

2. 使用`numpy.append()`添加列

numpy.append()函数的基本语法是：

numpy.append(arr, values, axis=None)

arr是要修改的数组。
values是要添加的值，这些值的形状必须匹配，以便可以沿指定的轴添加。
axis定义了沿哪个轴添加值。对于添加列，应该设置axis=1。

示例代码

下面是一些使用numpy.append()函数添加列的示例代码。

示例 1：向二维数组添加一列

import numpy as np

# 创建一个初始数组
arr = np.array([[1, 2], [3, 4]])

# 创建要添加的列
new_col = np.array([[5], [6]])

# 添加列
result = np.append(arr, new_col, axis=1)

print(result)

Output:

如何使用Numpy来向数组中添加列

示例 2：向二维数组添加多列

import numpy as np

# 创建一个初始数组
arr = np.array([[1, 2], [3, 4]])

# 创建要添加的多列
new_cols = np.array([[5, 9], [6, 10]])

# 添加列
result = np.append(arr, new_cols, axis=1)

print(result)

Output:

如何使用Numpy来向数组中添加列

示例 3：向空数组添加列

import numpy as np

# 创建一个空数组
arr = np.empty((0, 0), dtype=int)

# 创建要添加的列
new_col = np.array([[1], [2]])

# 添加列
result = np.append(arr, new_col, axis=1)

print(result)

示例 4：向数组添加列，其中列值来自计算

import numpy as np

# 创建一个初始数组
arr = np.array([[1, 2], [3, 4]])

# 计算要添加的列（每行的最大值）
new_col = np.max(arr, axis=1, keepdims=True)

# 添加列
result = np.append(arr, new_col, axis=1)

print(result)

Output:

如何使用Numpy来向数组中添加列

示例 5：向三维数组添加列

import numpy as np

# 创建一个三维数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 创建要添加的列
new_col = np.array([[[9], [10]], [[11], [12]]])

# 添加列
result = np.append(arr, new_col, axis=2)

print(result)