如何在Python中将大端24i24q格式的24bit采样转为32bit及complex64？

你的思路整体是正确的，但需要注意24bit有符号整数的解析这一关键细节——因为Numpy没有原生的24bit整数数据类型，直接读取无法直接映射，需要手动处理字节拼接和符号扩展。下面是具体的实现步骤和注意点：

核心步骤拆解 #

• 读取二进制文件：以二进制模式打开文件，读取全部字节。由于是大端字节序，每个I/Q采样各占3字节（24bit），总字节数应为采样数 × 6（I占3字节+Q占3字节）。
• 解析24bit有符号整数：
  1. 将读取的字节数组按3字节为一组拆分，分别对应I和Q通道的采样。
  2. 对每组3字节进行大端解析，扩展为32bit有符号整数（处理符号位：如果24bit的最高位是1，说明是负数，需要在高位补1到32bit）。
• 转换为float32并组合复数：将解析后的I/Q整数分别归一化到[-1.0, 1.0]范围转成float32，再通过numpy.complex64组合成复数数组。

示例代码 #

import numpy as np

def read_24i24q_to_complex64(file_path):
    # 读取全部字节
    with open(file_path, 'rb') as f:
        raw_bytes = f.read()
    
    # 转换为uint8数组，方便处理
    byte_array = np.frombuffer(raw_bytes, dtype=np.uint8)
    
    # 拆分I和Q的3字节组：每6字节对应一个复数采样（I:0-2, Q:3-5）
    i_bytes = byte_array.reshape(-1, 6)[:, 0:3]
    q_bytes = byte_array.reshape(-1, 6)[:, 3:6]
    
    # 将3字节大端24bit有符号整数转换为int32
    # 大端模式下，字节顺序是[高字节, 中字节, 低字节]
    i_int24 = (i_bytes[:, 0] << 16) | (i_bytes[:, 1] << 8) | i_bytes[:, 2]
    q_int24 = (q_bytes[:, 0] << 16) | (q_bytes[:, 1] << 8) | q_bytes[:, 2]
    
    # 符号扩展：24bit的最高位是第23位（从0开始计数），如果该位为1，说明是负数，需要补全高位
    mask = 1 << 23
    i_int32 = np.where(i_int24 & mask, i_int24 | (~((1 << 24) - 1)), i_int24).astype(np.int32)
    q_int32 = np.where(q_int24 & mask, q_int24 | (~((1 << 24) - 1)), q_int24).astype(np.int32)
    
    # 归一化到[-1.0, 1.0]，转float32
    max_val = 2**23 - 1
    i_float = i_int32.astype(np.float32) / max_val
    q_float = q_int32.astype(np.float32) / max_val
    
    # 组合成complex64数组
    complex_array = np.complex64(i_float + 1j * q_float)
    
    return complex_array

关键注意点 #

• 符号扩展：24bit有符号数的范围是[-2^23, 2^23 - 1]，如果直接把24bit无符号数转int32，负数会被当成大的正数，必须通过掩码判断最高位并补全高位1。
• 字节顺序：大端模式下，第一个字节是最高位字节，拼接时要注意移位顺序（高字节左移16位，中字节左移8位，低字节直接保留）。
• 归一化：必须用2^23 -1作为最大值，因为24bit有符号数的正数最大值是这个值，负数最小值是-2^23，这样归一化后能覆盖[-1.0, 1.0]区间。

内容的提问来源于stack exchange，提问作者Kokoloko