如何将字典转换为NumPy数组？

---

近年来，由于其多功能性和易用性，Python 赢得了极大的普及。其成功的一个关键原因是提供了许多库和包，简化了复杂的任务。NumPy 就是这样一个广泛用于科学计算和数据分析的库。它提供了一个强大的 N 维数组对象，可以高效地处理大型数据集并执行各种数学运算。

在本教程中，我们将学习如何使用 Python 将字典转换为 NumPy 数组。作为数据科学家和分析师，我们经常处理各种格式的数据，有时需要将数据从一种格式转换为另一种格式。在本文的下一节中，我们将探讨将字典转换为 NumPy 数组的分步过程，并讨论各种示例来演示转换过程。

如何将字典转换为NumPy数组？

在本节中，我们将讨论如何将字典转换为 NumPy 数组。我们将介绍将字典转换为 NumPy 数组的分步过程以及示例。

步骤 1：导入 NumPy 库

在我们开始之前，我们需要在 Python 中导入 NumPy 库。我们可以使用以下代码：

import numpy as np

在这里，我们导入了 NumPy 库并为其分配了别名“np”。这是在 Python 中使用 NumPy 的常见约定。

步骤 2：创建字典

为了本文的目的，我们将使用以下字典作为示例：

my_dict = {'a': 1, 'b': 2, 'c': 3, 'd': 4}

这是一个简单的字典，包含四个键值对。但是，我们可以使用将在后续步骤中讨论的方法将任何大小和形状的字典转换为 NumPy 数组。

步骤 3：将字典转换为 NumPy 数组

我们可以使用 `np.array()` 方法将字典转换为 NumPy 数组。`np.array()` 方法将可迭代对象作为输入，并返回 NumPy 数组。这是一个示例：

my_array = np.array(list(my_dict.items()))

在上面的代码片段中，我们首先使用 `items()` 方法将字典项转换为列表。然后，我们将此列表作为输入传递给 `np.array()` 方法以获取 NumPy 数组。

另一种将字典转换为 NumPy 数组的方法是使用 `np.asarray()` 方法。`np.asarray()` 方法也接受可迭代对象作为输入并返回 NumPy 数组。但是，如果输入对象已经是 NumPy 数组，它与 `np.array()` 方法的不同之处在于它不会创建输入对象的副本。这是一个示例：

my_array = np.asarray(list(my_dict.items()))

在此示例中，我们将字典项列表作为输入传递给 `np.asarray()` 方法以获取 NumPy 数组。

步骤 4：访问和操作 NumPy 数组

将字典转换为 NumPy 数组后，我们可以使用各种 NumPy 方法访问和操作数组。

要访问 NumPy 数组的元素，我们可以使用索引和切片。考虑一下访问 NumPy 数组项的以下代码：

print(my_array[0])    # Output: ['a' '1']
print(my_array[1][1]) # Output: 2

在第一行中，我们访问了 NumPy 数组的第一个元素，它是一个包含键值对 ('a', 1) 的一维数组。在第二行中，我们访问了 NumPy 数组的第二个元素，它也是一个包含键值对 ('b', 2) 的一维数组，然后访问该数组的第二个元素，即值 '2'。

就是这样！我们已成功地将字典转换为 NumPy 数组，并使用 NumPy 方法访问了该数组。

现在我们已经学习了如何将简单的字典转换为 NumPy 数组，让我们继续学习如何在 Python 中将嵌套字典转换为 NumPy 数组。

将嵌套字典转换为 NumPy 数组

在我们能够将嵌套字典转换为 NumPy 数组之前，我们需要创建一个嵌套字典示例。这是一个嵌套字典的示例：

my_dict = {
    "first": {"a": 1, "b": 2, "c": 3},
    "second": {"d": 4, "e": 5, "f": 6},
    "third": {"g": 7, "h": 8, "i": 9}
}

在此示例中，我们有一个嵌套字典，它包含三个键值对，其中每个值都是另一个字典。

要将嵌套字典转换为 NumPy 数组，我们首先需要使用 `values()` 方法从字典中提取值。然后，我们可以使用 `np.array()` 方法将值转换为 NumPy 数组。这是执行此操作的代码：

my_array = np.array(list(my_dict.values()))

在此代码中，我们使用 `list()` 方法将字典值转换为列表，然后将其传递给 `np.array()` 方法将其转换为 NumPy 数组。

让我们打印 NumPy 数组并查看其外观：

print(my_array)

输出

array([[[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]]])

如我们所见，嵌套字典已成功转换为 NumPy 数组。输出是一个三维数组，具有三行三列。

结论

在本教程中，我们学习了如何将 Python 字典转换为 NumPy 数组。我们讨论了创建字典、导入 NumPy 库和使用 NumPy 数组的分步过程。我们提供了示例，演示了如何使用 `np.array()` 和 `np.asarray()` 方法将简单和嵌套字典转换为 NumPy 数组。将字典转换为 NumPy 数组后，我们可以使用各种 NumPy 方法访问和操作数组。将字典转换为 NumPy 数组的过程在处理大型数据集并在其上执行数学运算时非常有用。