python - 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 .