Python中带小数部分二进制数的右移与位掩码实现方法

嘿，这个需求挺有意思的——Python原生位操作只针对整数，要处理带固定小数位的二进制数，得换个思路用定点数来玩，既符合Python风格，又能完美实现你要的右移和位掩码操作。我一步步给你拆解：

一、Python风格的定点数表示 #

你说的「1个整数位+2个小数位」本质是固定位数的定点数：把整个数的二进制当成一个整体整数存储，同时记录小数部分的位数。比如你例子里的0b101对应1.25，其实就是把1.25乘以2^2=4得到整数5（0b101），然后用「存储整数+小数位数」的组合来表示这个数。

用Python的类来封装是最清晰的方式，完全符合Python的面向对象风格：

class FixedPoint:
    def __init__(self, raw_int: int, fractional_bits: int):
        self.raw_int = raw_int  # 存储二进制对应的整数
        self.frac_bits = fractional_bits  # 小数部分的位数

    def to_decimal(self) -> float:
        """把定点数转换成十进制"""
        return self.raw_int / (2 ** self.frac_bits)

    def __repr__(self):
        return f"FixedPoint(raw=0b{self.raw_int:b}, frac_bits={self.frac_bits})"

测试一下你的例子：

# 0b101对应1整数位+2小数位，即1.25
fp = FixedPoint(0b101, 2)
print(fp.to_decimal())  # 输出1.25

二、右移操作的实现 #

你要的「右移1次得到0b010（0.5）」，对应到定点数就是直接对存储的raw_int执行整数右移，然后保持小数位数不变——因为右移整数等价于把整个定点数除以2^shift次数，正好对应你要的二进制移位效果。

给FixedPoint类加个右移方法：

def right_shift(self, shifts: int) -> 'FixedPoint':
    """执行右移操作，返回新的定点数"""
    return FixedPoint(self.raw_int >> shifts, self.frac_bits)

测试你的场景：

shifted_fp = fp.right_shift(1)
print(shifted_fp.raw_int)  # 输出2，即0b10
print(shifted_fp.to_decimal())  # 输出0.5，完美符合预期

这种方式比你之前用乘以2**-x的方法更高效，也更贴合「位操作」的直观逻辑。

三、位掩码获取特定位 #

要获取定点数的某一位或某几位，直接对raw_int用普通的整数位掩码操作就行——因为raw_int本身就是整个二进制数的整数形式，每一位的位置和你定义的二进制结构完全对应。

比如你例子里的0b101：

• 第2位（从右往左数，0是最低位）是整数位
• 第1位是小数第一位（权重0.5）
• 第0位是小数第二位（权重0.25）

给类加个获取特定位的方法，再扩展获取指定位段的功能：

def get_bit(self, bit_pos: int) -> int:
    """获取指定位置的位（0=最低位，对应最小的小数位）"""
    return (self.raw_int >> bit_pos) & 1

def get_bits(self, mask: int) -> int:
    """用掩码获取指定位段的整数结果"""
    return self.raw_int & mask

测试一下：

# 获取整数位（第2位）
print(fp.get_bit(2))  # 输出1
# 获取小数第一位（第1位）
print(fp.get_bit(1))  # 输出0
# 获取小数第二位（第0位）
print(fp.get_bit(0))  # 输出1

# 用掩码获取整个小数部分（掩码0b11=3）
print(fp.get_bits(0b11))  # 输出1，对应二进制01，即0.25
# 用掩码获取整数位（掩码0b100=4）
print(fp.get_bits(0b100))  # 输出4，对应二进制100，即1.0

总结一下 #

1. 表示方法：用定点数类封装「原始整数+小数位数」，清晰又符合Python面向对象风格；
2. 右移操作：直接对原始整数执行>>位操作，保持小数位数不变；
3. 位掩码：对原始整数用&掩码操作，轻松获取特定位或指定位段。

Python的任意精度整数特性还能帮你避免溢出问题，处理更多位数的定点数也毫无压力～

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