python 如何将元组列表转换为元组数组?

fcg9iug3  于 2022-12-17  发布在  Python
关注(0)|答案(3)|浏览(243)

我列了一个清单:

l=[(1,2),(3,4)]

我想把它转换成一个numpy数组,并保持数组项类型为tuple:

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

但是numpy.array(l)将给予:

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

并且项类型已从tuple更改为numpy.ndarray,则我指定了项类型

numpy.array(l,numpy.dtype('float,float'))

这给出:

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

但项类型不是元组而是numpy.void,所以问题是:

how to convert it to a numpy.array of tuple,not of numpy.void?
m528fe3b

m528fe3b1#

你可以有一个object dtype的数组,让数组的每个元素都是一个元组,如下所示-

out = np.empty(len(l), dtype=object)
out[:] = l

样品运行-

In [163]: l = [(1,2),(3,4)]

In [164]: out = np.empty(len(l), dtype=object)

In [165]: out[:] = l

In [172]: out
Out[172]: array([(1, 2), (3, 4)], dtype=object)

In [173]: out[0]
Out[173]: (1, 2)

In [174]: type(out[0])
Out[174]: tuple
xtupzzrd

xtupzzrd2#

由于某种原因,如果您只需要一行代码,您就不能简单地这样做(即使Divakar的答案最终留给您的是dtype=object):

np.array([(1,2),(3,4)], dtype=object)

相反,您必须这样做:

np.array([(1,2),(3,4)], dtype="f,f")

"f,f"向数组发出信号,表示它正在接收两个浮点数的元组(或者您可以使用"i,i"表示整数)。如果需要,您可以通过将.astype(object)添加到上一行的末尾来强制转换回一个对象。

g52tjvyc

g52tjvyc3#

刚刚发现panda有办法解决这个问题。你可以使用他们的MultiIndex类来创建一个元组数组,因为所有panda索引都是 Package 的一维numpy数组。这就像在元组列表上调用Index构造函数一样简单。

>>> import pandas as pd
>>> tups = [(1, 2), (1, 3)]
>>> tup_array = pd.Index(tups).values
>>> print(type(tup_array), tup_array)
<class 'numpy.ndarray'> [(1, 2) (1, 3)]

相关问题