numpy.array工作不正常？

Question

我正在尝试将列表列表转换为数组,并且发生了一些奇怪的事情.

例如,如果我将列表声明为:

a=[]
a.append(['a',1,10,100])
a.append(['b',2,20,200])
a.append(['c',3,30,300])
a.append(['d',4,40,400])

然后转换为矢量

a=np.array(a)

最终的结果是

[['a','1','1','1'],['b','2','2','2'],['c','3','3','3'],['d','4','4','4']]

我是一个完全的Python初学者,但是从我通过使用np.array读到的内容中,向量中的所有内容都被转换为字符串(如果我错了,请纠正我).为什么在这里忽略零,我该怎么做才能修复它？

Answer 1

由于您的数组具有非统一数据类型,因此必须指定"结构化" dtype

In [2]: a = [('a', 1, 10, 100),
   ...:      ('b', 2, 20, 200),
   ...:      ('c', 3, 30, 300),
   ...:      ('d', 4, 40, 400)]

In [3]: a = np.array(a, dtype = "S1, int, int, int")

In [4]: a
Out[4]: 
array([('a', 1, 10, 100),
       ('b', 2, 20, 200),
       ('c', 3, 30, 300),
       ('d', 4, 40, 400)], 
      dtype=[('f0', 'S1'), ('f1', '<i8'), ('f2', '<i8'), ('f3', '<i8')])

您可以轻松访问哪些:

In [5]: a.dtype.names = ['name','x','y','z']

In [6]: a['x']
Out[6]: array([1, 2, 3, 4])

In [7]: a['name']
Out[7]: 
array(['a', 'b', 'c', 'd'], 
      dtype='|S1')

In [8]: a[0]
Out[8]: ('a', 1, 10, 100)