Numpy：从字典创建NumPy数组的最佳方法

在本文中，我们将介绍使用Python库Numpy从字典中创建NumPy数组的最佳方法。NumPy是一种基于Python的科学计算库，用于在Python中进行高级数学、科学和工程计算。使用NumPy，我们可以在Python中创建和操作多维数组及其数学运算。在数据科学、机器学习等领域，NumPy是必不可少的工具。

如果你的数据已经存储在Python字典中，那么将其转换为NumPy数组是一个常见的任务。下面，让我们看看如何使用NumPy实现这个操作。

阅读更多：Numpy 教程

字典中的值为元组

当字典中的值为元组时，可以使用np.array()函数创建NumPy数组，例如：

import numpy as np

my_dict = {'a': (1, 2), 'b': (3, 4), 'c': (5, 6)}
my_array = np.array(list(my_dict.values()))

print(my_array)

输出：

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

这里我们使用了list()函数将字典的值转换为列表。然后，我们使用numpy的np.array()函数将列表转换为NumPy数组。

字典中的值为列表

当字典中的值为列表时，可以使用np.array()函数和np.vstack()函数创建NumPy数组，例如：

import numpy as np

my_dict = {'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9]}
my_array = np.array(list(my_dict.values()))
my_array = np.vstack(my_array)

print(my_array)

输出：

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

在这个例子中，我们首先使用list()和np.array()函数将字典的值转换为NumPy数组。然后，我们使用np.vstack()函数将数组沿着垂直方向堆叠成一个NumPy数组。

字典中的键和值

如果字典中的键和值都需要作为NumPy数组的一部分，例如：

import numpy as np

my_dict = {'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9]}
my_array = np.array(list(my_dict.items())).reshape(-1, 2)

print(my_array)

输出：

array([['a', '1,2,3'],
       ['b', '4,5,6'],
       ['c', '7,8,9']], dtype='<U21')

在这个例子中，我们使用list()和np.array()函数将字典的键和值转换为NumPy数组，并使用reshape()函数将数组重新格式化为所需的形状。

总结

使用Numpy从Python字典中创建NumPy数组是一项非常常见的任务。在字典中的值为元组时，可以使用np.array()函数创建NumPy数组。当字典中的值为列表时，可以使用np.vstack()函数。如果需要包含键和值，可以使用np.array()函数和reshape()函数进行转换。

希望这篇文章对您创建NumPy数组非常有帮助。感谢阅读！