Numpy 为数组添加行/列标题

NumPy是基于Python的开源数值计算库，广泛应用在数据科学以及机器学习领域中。处理数据时，我们通常需要为数据集添加行/列标题，以方便理解和处理数据。本文将介绍如何为NumPy数组添加行/列标题，并通过示例说明。

阅读更多：Numpy 教程

创建NumPy数组

首先，我们需要创建一个NumPy数组来演示如何为其添加行/列标题。我们可以使用numpy.array()函数来创建一个数组，并传入一个列表（List）类型的数据作为参数，如下所示：

import numpy as np

data = [[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9],
        [10, 11, 12]]

np_array = np.array(data)
print(np_array)

输出结果如下：

[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]

添加行/列标题

为了方便理解数据，我们需要为NumPy数组添加行/列标题。我们可以使用numpy.insert()函数来插入数据，以添加行/列标题。我们可以使用numpy.shape属性来获取数组的形状，以确定在哪个位置插入行/列标题。下面是添加行/列标题的代码：

# 添加行标题
row_title = np.array(['A', 'B', 'C', 'D']).reshape(4, 1)
np_array = np.insert(np_array, 0, row_title, axis=1)
print(np_array)

# 添加列标题
col_title = np.array(['X', 'Y', 'Z'])
np_array = np.insert(np_array, 0, col_title, axis=0)
print(np_array)

输出结果如下：

[['X' '1' '2' '3']
 ['Y' '4' '5' '6']
 ['Z' '7' '8' '9']
 [' ' '10' '11' '12']]

[['X' 'Y' 'Z']
 ['A' '1' '2' '3']
 ['B' '4' '5' '6']
 ['C' '7' '8' '9']
 ['D' '10' '11' '12']]

我们可以看到，行标题被添加到了列之前，列标题被添加到了行之前。

提取行/列数据

我们可以使用numpy的切片功能获取NumPy数组的行/列数据。我们可以使用以下代码来获取数据：

# 获取特定行
row_data = np_array[1]
print(row_data)

# 获取特定列
col_data = np_array[:, 1]
print(col_data)

输出结果如下：

['Y' '4' '5' '6']
['Y' '1' '4' '7' '10']

总结

本文介绍了如何为NumPy数组添加行/列标题，并通过示例向读者演示了如何提取行/列数据。使用这些技巧，我们可以更方便地理解和处理数据。