从字典创建NumPy数组的最佳方法是使用numpy.array()
函数,并将字典转换为列表列表。以下是一个示例:
import numpy as np
# 定义一个字典
data = {'column1': [1, 2, 3], 'column2': [4, 5, 6], 'column3': [7, 8, 9]}
# 将字典转换为列表列表
data_list = [list(value) for value in data.values()]
# 创建NumPy数组
array = np.array(data_list)
# 转置数组以获得所需形状
array = array.T
# 将列标签添加为数组字段
array = np.core.records.fromarrays(array.transpose(), names=[key for key in data.keys()])
print(array)
输出:
[(1, 4, 7) (2, 5, 8) (3, 6, 9)]
在这个示例中,我们首先导入了NumPy库,然后创建了一个字典data
。接下来,我们将字典转换为列表列表data_list
。然后,我们使用numpy.array()
函数创建一个NumPy数组,并将其转置以获得所需的形状。最后,我们使用numpy.core.records.fromarrays()
函数将列标签添加为数组字段。
领取专属 10元无门槛券
手把手带您无忧上云