将列表列表转换为具有多种数据类型的numpy数组

时间:2015-08-31 05:14:57

标签: arrays numpy types

我有一个我从文件中读取的列表列表。每个内部列表的长度为六个元素,并且有3个字符串和5个浮点数。如何将此列表列表转换为numpy数组?谢谢!

2 个答案:

答案 0 :(得分:3)

你想要一个结构化数组,一个具有化合物QAbstractItemModel

的数组

列表的示例列表:

dtype

尝试创建一个常规数组,生成一个字符串数组:

In [4]: ll = [['one','two',1,1.23],['four','five',4,34.3],['six','seven',4,34.3]]

但是如果我指定一个In [5]: np.array(ll) Out[5]: array([['one', 'two', '1', '1.23'], ['four', 'five', '4', '34.3'], ['six', 'seven', '4', '34.3']], dtype='|S5') 包含2个字符串,并且int和一个浮点数,我得到一个1d结构化数组:

dtype

请注意,我必须将内部列表转换为元组。这是结构化数组如何获取其输入,以及它如何显示它。它有助于区分结构化的行'从制服'行'常规(2d)阵列。

这是In [8]: np.array([tuple(x) for x in ll],dtype='S5,S5,i,f') Out[8]: array([('one', 'two', 1, 1.2300000190734863), ('four', 'five', 4, 34.29999923706055), ('six', 'seven', 4, 34.29999923706055)], dtype=[('f0', 'S5'), ('f1', 'S5'), ('f2', '<i4'), ('f3', '<f4')]) genfromtxtloadtxt文件中读取时产生的结构化数组。

还有其他方法可以指定csv,还有其他一些方法可以将数据加载到这样的数组中。但这是一个开始。

进一步测试https://stackoverflow.com/a/47774915/901925表明此元组转换不是那么耗时。简单地创建阵列需要更多时间。

答案 1 :(得分:0)

我遇到了同样的问题,但是元组无法解决。所以我发现(python 3.7.1):

ll = [['one','two',1,1.23],['four','five',4,34.3],['six','seven',4,34.3]]

np.array(ll, dtype = 'object')

结果:

array([['one', 'two', 1, 1.23],
   ['four', 'five', 4, 34.3],
   ['six', 'seven', 4, 34.3]], dtype=object)