首頁  >  文章  >  後端開發  >  Go 中如何將位元組轉換為 float32 陣列?

Go 中如何將位元組轉換為 float32 陣列?

王林
王林轉載
2024-02-06 08:33:08881瀏覽

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
}

更新參考評論:

我對您收到的內容仍然有點困惑,所以讓我們嘗試一下這兩種可能性。

如果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中文網其他相關文章!

陳述:
本文轉載於:stackoverflow.com。如有侵權,請聯絡admin@php.cn刪除