numpy 加一列|极客教程

numpy 加一列

在进行数据处理和分析时，我们经常需要向已有的数据集中添加新的列。使用NumPy库，我们可以很方便地实现添加新列的操作。本文将详细介绍如何使用NumPy向数组中加入一列数据。

创建一个数组

首先，让我们创建一个示例数组作为我们的数据集。假设我们有一个包含学生姓名和分数的数组。

import numpy as np

# 创建一个包含学生姓名和分数的数组
students = np.array([['Alice', 85],
                     ['Bob', 72],
                     ['Cathy', 90],
                     ['David', 88]])

print("原始数组：")
print(students)

运行以上代码，我们可以得到以下输出：

原始数组：
[['Alice' '85']
 ['Bob' '72']
 ['Cathy' '90']
 ['David' '88']]

现在，我们已经有了一个包含学生姓名和分数的数组students。

添加一列数据

接下来，我们将向这个数组中添加一列数据，比如性别信息。我们可以使用np.concatenate()函数将原始数组和新的一列数据进行合并。

# 创建一个性别数组
gender = np.array([['Female'],
                   ['Male'],
                   ['Female'],
                   ['Male']])

# 将性别数组和学生数组进行合并
students = np.concatenate((students, gender), axis=1)

print("添加性别信息后的数组：")
print(students)

运行以上代码，我们可以得到以下输出：

添加性别信息后的数组：
[['Alice' '85' 'Female']
 ['Bob' '72' 'Male']
 ['Cathy' '90' 'Female']
 ['David' '88' 'Male']]

通过以上代码，我们成功向原始数组中添加了一列性别信息。

除了使用np.concatenate()函数外，我们还可以使用np.hstack()和np.column_stack()函数来实现向数组添加新列的操作。这里以np.column_stack()为例：

# 创建一个城市数组
city = np.array([['New York'],
                 ['Los Angeles'],
                 ['Chicago'],
                 ['Houston']])

# 将城市数组和学生数组进行列合并
students = np.column_stack((students, city))

print("添加城市信息后的数组：")
print(students)

运行以上代码，我们可以得到以下输出：

添加城市信息后的数组：
[['Alice' '85' 'Female' 'New York']
 ['Bob' '72' 'Male' 'Los Angeles']
 ['Cathy' '90' 'Female' 'Chicago']
 ['David' '88' 'Male' 'Houston']]