给NumPy数组添加列|极客笔记

给NumPy数组添加列

NumPy是一个功能强大的Python库，用于处理大型、多维数组和矩阵。在进行数据处理和科学计算时，我们经常需要修改NumPy数组的形状和结构。本文将详细介绍如何向NumPy数组添加列。

1. 创建NumPy数组

在开始添加列之前，我们先创建一个示例的NumPy数组。可以使用NumPy库中的array函数来创建一个NumPy数组。下面是一个示例：

import numpy as np

# 创建一个2x3的二维数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
print('原始数组：')
print(array_2d)

输出结果为：

给NumPy数组添加列

2. 使用`np.column_stack`函数添加列

np.column_stack函数可以将给定的一维数组插入到二维数组的列中。下面是一个示例：

import numpy as np

# 创建一个2x3的二维数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])

# 创建一个一维数组
column = np.array([7, 8])

# 使用np.column_stack函数添加列
result = np.column_stack((array_2d, column))

print('添加列后的数组：')
print(result)

输出结果为：

给NumPy数组添加列

3. 使用`np.hstack`函数添加列

np.hstack函数可以水平拼接数组，可以将一维数组添加为一个列向量。下面是一个示例：

import numpy as np

# 创建一个2x3的二维数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])

# 创建一个一维数组
column = np.array([7, 8])

# 使用np.hstack函数添加列
result = np.hstack((array_2d, column[:, np.newaxis]))

print('添加列后的数组：')
print(result)

输出结果为：

给NumPy数组添加列

4. 使用`np.concatenate`函数添加列

np.concatenate函数可以按照指定的轴将多个数组进行合并。使用axis参数指定合并的轴方向，当axis=1时，表示按列进行合并。下面是一个示例：

import numpy as np

# 创建一个2x3的二维数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])

# 创建一个一维数组
column = np.array([7, 8])

# 使用np.concatenate函数添加列
result = np.concatenate((array_2d, column[:, np.newaxis]), axis=1)

print('添加列后的数组：')
print(result)

输出结果为：

给NumPy数组添加列