Go语言中将字节流转换为Float32数组的实用指南

package main

import (
    "encoding/binary"
    "encoding/hex" // 稍后用于处理十六进制字符串
    "fmt"
    "math"
)

// BytesFloat32 将4字节切片转换为float32
// 假设输入字节流是小端序（Little-Endian）
func BytesFloat32(bytes []byte) float32 {
    // 使用 binary.LittleEndian.Uint32 解析4个字节为uint32
    // 它会按照小端序规则从字节切片中读取并组合成一个uint32
    bits := binary.LittleEndian.Uint32(bytes)
    // 将uint32的位模式直接转换为float32
    f := math.Float32frombits(bits)
    return f
}

// GetFloatArray 将字节切片转换为float32数组
// 假定每个float32占用4字节
func GetFloatArray(aBytes []byte) []float32 {
    // 计算预期的float32数量。如果字节长度不是4的倍数，则会忽略末尾不足4字节的部分。
    numFloats := len(aBytes) / 4
    // 创建一个足够大的float32切片
    aArr := make([]float32, numFloats)

    for i := 0; i < numFloats; i++ {
        // 每次取4个字节进行转换
        start := i * 4
        end := start + 4
        // 确保切片操作不会越界，尽管numFloats的计算已经确保了这一点，
        // 但在更复杂的逻辑中，显式检查仍是好习惯。
        if end > len(aBytes) {
            // 理论上不会发生，因为numFloats已经考虑了长度
            fmt.Printf("警告: 字节数据长度不足以构成完整的float32数组，预期 %d 字节，实际 %d 字节\n", (i+1)*4, len(aBytes))
            break // 退出循环，返回已转换的部分
        }
        aArr[i] = BytesFloat32(aBytes[start:end])
    }
    return aArr
}

func main() {
    // 假设这是从Redis获取的原始字节字符串
    // Python numpy.array([1.1, 2.2, 3.3], dtype=np.float32).tobytes() 产生的字节流
    // 对应十六进制: 3F8CCCCD400CCCCD40533333
    // 注意Python输出的字节序通常是小端序，因此1.1对应的十六进制是3F8CCCCD，
    // 但在内存中按小端序存储为CD CC 8C 3F。
    var aBytesStr string = "\xcd\xcc\x8c?\xcd\xcc\x0c@33S@"
    fmt.Println("--- 场景一：从原始字节字符串转换 ---")
    fmt.Printf("原始字符串: %q\n", aBytesStr)
    floatArrayFromBytesStr := GetFloatArray([]byte(aBytesStr))
    fmt.Printf("转换结果: %v\n", floatArrayFromBytesStr) // 预期: [1.1 2.2 3.3]
}