为什么从列表中创建float32数组会引发值错误

Question

我正试图创建一个float32数组，如下所示，

np.float32([kp2, kp2+dir_v*dist, kp2 + dir_v_r*dist])

然而，我得到了以下错误，

ValueError：用序列设置数组元素。

我试图如下所示将其显式转换为numpy数组，然后重试，但仍然收到相同的错误

np.array([kp2, kp2+dir_v*dist, kp2 + dir_v_r*dist]).astype(float)

我错过了什么？

Answer 1

使用dtype参数指定np.array()的数据类型。

np.array([kp2, kp2+dir_v*dist, kp2 + dir_v_r*dist], dtype=float)

Answer 2

下面是一个产生错误的示例：

In [910]: np.array([1,[1,2,3],4])                                                              
Out[910]: array([1, list([1, 2, 3]), 4], dtype=object)

在没有float转换的情况下很好，但随后出现错误：

In [911]: np.array([1,[1,2,3],4]).astype(float)                                                
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
TypeError: float() argument must be a string or a number, not 'list'
The above exception was the direct cause of the following exception:
ValueError                                Traceback (most recent call last)
<ipython-input-911-9c007059d6ad> in <module>
----> 1 np.array([1,[1,2,3],4]).astype(float)
ValueError: setting an array element with a sequence.

我怀疑[kp2, kp2+dir_v*dist, kp2 + dir_v_r*dist]包含一个数字、数组和列表的组合，从而生成一个对象数据类型数组。

如果列表确实包含数字和数字列表(或数组(的混合，那么hstack可能就是您想要的：

In [922]: np.hstack([1,[1,2,3],4])                                                             
Out[922]: array([1, 1, 2, 3, 4])