我有一个这样的清单:
l=[(1,2),(3,4)]
我想将它转换为numpy数组,并将数组项类型保持为元组:
array([(1,2),(3,4)])
但是numpy.array(l)会给出:
array([[1,2],[3,4)]])
并且项目类型已从元组更改为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?
答案 0 :(得分:5)
你可以有一个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
答案 1 :(得分:0)
出于某种原因,如果您正在寻找一行代码,则不能简单地执行此操作(即使 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)
来转换回对象)。