numpy中的matlab single()函数？

Question

我尝试将 matlab 代码转换为 python/numpy 代码。

我有这行：

l = single(l)

“l”是一个数组数组，正如 matlab 文档所说的“转换为单精度”。

我怎样才能用 numpy 做到这一点？

Answer 1

要将二维 numpy 数组转换为单精度数组，请使用 astype 并为其指定 float32 参数。例如：

>>> import numpy as np
>>> a = np.array([[1.], [2.], [3.]])
>>> a
array([[ 1.],
       [ 2.],
       [ 3.]])
>>> a = a.astype('float32')
>>> a
array([[ 1.],
       [ 2.],
       [ 3.]], dtype=float32)

有关数值和数组数据类型的更多信息，请参阅the documentation。