Python中将内置数据类型int/float替换为随机变量的实现方案咨询

你之前尝试子类化int的方案无法生效的原因是：int是不可变类型，实例创建时的取值就已经固定，后续任何操作都无法修改它的内置存储值，覆写__call__也不会在直接运算/返回变量时触发。

核心实现思路是自定义类覆写数值相关的魔术方法/使用描述符协议，实现每次访问时动态生成随机值，完全适配现有代码的调用逻辑，不需要修改原有业务代码。

方案1：通用随机变量类（适配所有场景） #

该方案实现的随机变量可以直接替换原有代码中的int/float固定值，不需要修改任何调用逻辑：

import random

class StochasticVariable:
    def __init__(self, min_val=0, max_val=100, value_type=int):
        self.min_val = min_val
        self.max_val = max_val
        self.value_type = value_type

    def _get_random_val(self):
        """每次访问时动态生成随机值"""
        if self.value_type is int:
            return random.randint(self.min_val, self.max_val)
        elif self.value_type is float:
            return random.uniform(self.min_val, self.max_val)
        raise TypeError("仅支持int/float两种数值类型")

    # 覆写数值运算魔术方法
    def __add__(self, other):
        return self._get_random_val() + other
    def __radd__(self, other):
        return other + self._get_random_val()
    def __sub__(self, other):
        return self._get_random_val() - other
    def __rsub__(self, other):
        return other - self._get_random_val()
    def __mul__(self, other):
        return self._get_random_val() * other
    def __rmul__(self, other):
        return other * self._get_random_val()
    def __truediv__(self, other):
        return self._get_random_val() / other
    def __rtruediv__(self, other):
        return other / self._get_random_val()

    # 覆写类型转换方法
    def __int__(self):
        return int(self._get_random_val())
    def __float__(self):
        return float(self._get_random_val())

    # 覆写打印/返回逻辑
    def __repr__(self):
        return repr(self._get_random_val())
    def __str__(self):
        return str(self._get_random_val())

效果验证 #

普通函数场景

def existing_function(v):
    return v + 1

# 替换原有固定值v=1为随机变量
v = StochasticVariable(min_val=1, max_val=10)
print(existing_function(v)) # 输出2~11之间的随机值
print(existing_function(v)) # 两次输出结果不同

类属性场景

class X:
    def __init__(self, v):
        self.v = v
    def existing_function(self):
        return self.v

o = X(v)
print(o.existing_function()) # 输出1~10之间的随机值
print(o.existing_function()) # 两次输出结果不同

方案2：类属性专用描述符实现 #

如果仅需要在类属性层面实现随机取值，使用描述符协议的实现更轻量：

import random

class StochasticAttribute:
    def __init__(self, min_val=0, max_val=100, value_type=int):
        self.min_val = min_val
        self.max_val = max_val
        self.value_type = value_type

    def __get__(self, instance, owner):
        if self.value_type is int:
            return random.randint(self.min_val, self.max_val)
        return random.uniform(self.min_val, self.max_val)

# 使用示例
class X:
    # 直接定义为类属性，不需要在__init__中赋值
    v = StochasticAttribute(min_val=1, max_val=10)
    def existing_function(self):
        return self.v

o = X()
print(o.existing_function()) # 每次调用返回不同随机值
print(o.existing_function())

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