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)

有关数字和数组数据类型的更多信息，请参阅文档。