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

NumPy 是一个基于Python的科学计算库，它支持多维数组和矩阵运算，并提供了许多计算函数。在Python编程中，我们通常会使用字典来存储键值对，但是在NumPy中，数组是最基本的数据结构。因此，需要在某些情况下将字典转换为NumPy数组。在本文中，我们将介绍如何将字典转换为NumPy数组。

使用numpy.array()方法

使用numpy.array()方法可以将字典转换为NumPy数组。此方法需要将字典作为输入，并使用dtype参数指定数组数据类型。值得注意的是，字典的键将作为数组的索引，因此需要将字典的值列表转换为NumPy数组。

import numpy as np

# 定义字典
my_dict = {'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9]}

# 将字典转换为NumPy数组
my_array = np.array(list(my_dict.values()), dtype=int)

print(my_array)

输出：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

在上面的代码中，我们首先定义了一个字典my_dict，它包含三个键值对。然后，我们使用numpy.array()方法将字典转换为NumPy数组，并将dtype参数设置为int，以指定数组的数据类型。最后，我们打印这个数组，输出为一个三行三列的矩阵。

使用numpy.fromiter()方法

另一种将字典转换为NumPy数组的方法是使用numpy.fromiter()方法。此方法需要一个迭代器作为输入。我们可以使用字典的值列表创建一个迭代器，并使用dtype参数指定数组数据类型。键可以通过使用numpy.arange()方法来创建。

import numpy as np

# 定义字典
my_dict = {'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9]}

# 将字典转换为NumPy数组
my_array = np.fromiter((value for key in my_dict for value in my_dict[key]), 
                        dtype=int).reshape(len(my_dict),-1)

print(my_array)

输出：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

在上面的代码中，我们首先定义了一个字典my_dict，它包含三个键值对。然后，我们使用numpy.fromiter()方法将字典转换为NumPy数组，并将dtype参数设置为int，以指定数组的数据类型。最后，我们打印这个数组，输出为一个三行三列的矩阵。

结论

如何将字典转换为NumPy数组？有两种方法可以实现。第一种方法是使用numpy.array()方法，它需要将字典转换为值列表，并指定数组数据类型。这种方法将使用字典的键作为数组的索引。另一个方法是使用numpy.fromiter()方法，它需要值列表迭代器和数组数据类型作为输入，可以使用numpy.arange()方法创建键。两个方法都可以将字典转换为NumPy数组。