Numpy 数组添加列名称

阅读更多：Numpy 教程

导言

在本文中，我们将介绍如何在NumPy数组中添加列名称。NumPy是一个Python库，它提供了一个大量用于科学计算的多维数组对象。在NumPy数组中，我们可以使用数字、布尔值、字符串和对象等数据类型。NumPy数组具有以下优点：

• 性能优越：NumPy是基于C语言编写的，因此比Python列表更快。
• 广泛的数学函数：NumPy拥有许多用于线性代数、傅立叶变换和随机数生成等方面的功能，这些功能对科学计算非常重要。
• 大数据集处理：NumPy可以处理数百万甚至数十亿个元素的数组，而Python默认列表的处理速度会随着元素数量的增加而降低。

有时，在使用NumPy数组进行数据分析时，我们需要为数组的每一列添加列名。例如，我们可以有一个包含学生名称、年龄、性别和分数的数组。在这种情况下，我们可能想为每列添加名称，以使数据更加直观易读。

简单的NumPy数组添加列名称

要添加NumPy数组的列名称，请先创建数组，然后使用以下代码将名称添加到数组中:

import numpy as np

# 创建2列带有4个项的数组
a = np.array([[1, 2], [3, 4], [5, 6], [7, 8]])

# 添加列名称
a = np.insert(a, 0, values=["col1", "col2"], axis=1)

print(a)

输出结果：

[['col1' 'col2' '1' '2']
 ['col1' 'col2' '3' '4']
 ['col1' 'col2' '5' '6']
 ['col1' 'col2' '7' '8']]

在上述代码中，我们首先创建一个2列4行的NumPy数组。接下来，我们使用np.insert函数将值（即列名）插入到数组的第0列（axis=1）。最后，我们打印结果以确认列名是否已成功添加到数组中。

使用结构化数组添加列名称

另一种添加NumPy数组列名称的方式是使用结构化数组。结构化数组是一种特殊的NumPy数组，它允许我们对数据进行更灵活的处理。要创建结构化数组，请使用以下代码：

import numpy as np

# 创建一个结构化数组
a = np.zeros((4,), dtype=[("Name", "U10"), ("Age", "i4"), ("Gender", "U1"), ("Score", "f4")])

# 添加值
a["Name"] = ["Adam", "Bob", "Chris", "David"]
a["Age"] = [20, 21, 22, 23]
a["Gender"] = ["M", "M", "M", "M"]
a["Score"] = [95.5, 92.3, 87.6, 90.0]

print(a)

输出结果：

[('Adam', 20, 'M', 95.5) ('Bob', 21, 'M', 92.3) ('Chris', 22, 'M', 87.6)
 ('David', 23, 'M', 90. )]

在上述代码中，我们首先使用np.zeros函数创建一个无元素的结构化数组，通过在dtype参数中指定每个列的名称和数据类型。接下来，我们为每列添加了值。请注意，我们可以使用列名称来访问相应的列。

总结

在上文中，我们介绍了两种在NumPy数组中添加列名称的方式：简单添加和结构化数组。我们可以根据情况选择不同的方法。无论使用哪种方法，添加列名称可以使NumPy数组更直观易读，更易于进行数据分析。