Go 中如何将字节转换为 float32 数组？-Golang-PHP中文网

Go 中如何将字节转换为 float32 数组？

王林

发布： 2024-02-06 08:33:08

转载

1097人浏览过

go 中如何将字节转换为 float32 数组？

问题内容

我正在将一个 float32 数字数组以字节格式从 Python 脚本写入 Elasticache Redis 集群，然后在 Go 脚本中从 Elasticache 读取字节（作为字符串）。如何在 Go 脚本中将 bytes-as-string 转换回原始 float32 数组？

Python 示例：

import numpy as np
import redis
a = np.array([1.1, 2.2, 3.3], dtype=np.float32)
a_bytes = a.tobytes(order="C") #I have also tried order="F" with no luck
print(a_bytes) #Output: b'\xcd\xcc\x8c?\xcd\xcc\x0c@33S@'
redis_client = redis.cluster.RedisCluster(host=<elasticache config endpoint>, port=6379)
redis_client.mset_nonatomic({"key1": a_bytes})

登录后复制

这是我在 Go（游乐场）中尝试过的示例

package main

import (
    "fmt"
    "math"
    "strconv"
)

func main() {
    // aBytesStr is an example value retrieved from Elasticache
    // aBytesStr is type string, not raw bytes
    var aBytesStr string = "\xcd\xcc\x8c?\xcd\xcc\x0c@33S@"
    aHex := fmt.Sprintf("%X", aBytesStr)
    fmt.Println(aHex) // Output: CDCC8C3FCDCC0C4033335340
    var aArr [3]float32
    for i := 0; i < 3; i++ {
        aHex1 := aHex[i*8 : i*8+8]
        aParsed, err := strconv.ParseUint(aHex1, 16, 32)
        if err != nil {
            return
        }
        aArr[i] = math.Float32frombits(uint32(aParsed))
    }
    fmt.Println(aArr)
    // Expected output: [1.1 2.2 3.3]
    // Actual output [-4.289679e+08 -4.2791936e+08 4.17524e-08]
}

登录后复制

正确答案

您使用的示例代码是“转换十六进制，表示为字符串”；你有原始字节（我认为基于 aHex：CDCC8C3FCDCC0C4033335340），因此直接转换更简单（虽然你可以将字节转换为十六进制字符串，然后进行转换，这样做只会增加不必要的工作/复杂性）。

从这个答案中我们得到（游乐场):

func GetFloatArray(aBytes []byte) []float32 {
    aArr := make([]float32, 3)
    for i := 0; i < 3; i++ {
        aArr[i] = BytesFloat32(aBytes[i*4:])
    }
    return aArr
}

func BytesFloat32(bytes []byte) float32 {
    bits := binary.LittleEndian.Uint32(bytes)
    float := math.Float32frombits(bits)
    return float
}

登录后复制

更新参考评论：

怪兽AI数字人

数字人短视频创作，数字人直播，实时驱动数字人

查看详情

我对您收到的内容仍然有点困惑，所以让我们尝试一下这两种可能性。

如果redis查询返回原始数据（字节）作为go字符串（即"\xcd\xcc\x8c?\xcd\xcc\x0c@33S@"），那么你可以将转换为 []byte (游乐场)

func main() {
    var aBytesStr string = "\xcd\xcc\x8c?\xcd\xcc\x0c@33S@"
    fmt.Println(GetFloatArray([]byte(aBytesStr)))
}

登录后复制

如果 Redis 返回包含 ASCII (/UTF-8) 表示形式的字符串（即 CDCC = []byte{0x41, 0x44, 0x43, 0x43}），最简单的方法可能是使用 encoding/hex 对此进行解码（游乐场)

func main() {
    aHex := "CDCC8C3FCDCC0C4033335340"
    b, err := hex.DecodeString(aHex)
    if err != nil {
        panic(err)
    }
    fmt.Println(GetFloatArray(b))
}

登录后复制

请注意，您原来的方法可能有效，但是，正如上面评论中指出的，您需要处理 Endianness 因此以下内容将起作用（playground - 你可以提高效率，我的目的是为了清晰）：

byte1 := aHex[i*8 : i*8+2]
byte2 := aHex[i*8+2 : i*8+4]
byte3 := aHex[i*8+4 : i*8+6]
byte4 := aHex[i*8+6 : i*8+8]
aParsed, err := strconv.ParseUint(byte4+byte3+byte2+byte1, 16, 32)

登录后复制

但是，这对运行代码的 CPU 做出了假设，这意味着前面的答案更可取。

以上就是Go 中如何将字节转换为 float32 数组？的详细内容，更多请关注php中文网其它相关文章！