如何使用JSON序列化Python伪随机数生成器（PRNG）的状态？

random.getstate()返回的状态是嵌套的元组结构，包含整数和内层元组。但JSON序列化时会自动将所有元组转为列表，即便你把外层转成tuple(restored_state)，内层的列表也无法变回元组，而random.setstate()要求整个状态向量的所有嵌套结构都必须是元组类型，这就是报错的核心原因。

方法1：使用pickle（最简单直接） #

如果只是为了保存和恢复PRNG状态，pickle是Python原生序列化工具，能完美保留状态的类型结构：

import random
import pickle

if __name__ == '__main__':
    random.seed()
    # 保存状态
    random_state = random.getstate()
    with open("random_state.pkl", "wb") as f:
        pickle.dump(random_state, f)
    
    # 恢复状态
    with open("random_state.pkl", "rb") as f:
        restored_state = pickle.load(f)
    random.setstate(restored_state)

方法2：适配JSON格式（满足你的学习需求） #

要让JSON正确处理元组，需要自定义序列化和反序列化逻辑，给元组添加识别标记，反序列化时再转换回去：

import random
import json

def tuple_encoder(obj):
    """把元组转换成带标记的字典，让JSON可识别"""
    if isinstance(obj, tuple):
        return {"__type__": "tuple", "data": list(obj)}
    raise TypeError(f"Object of type {type(obj).__name__} is not JSON serializable")

def tuple_decoder(obj):
    """把标记的字典转回元组"""
    if "__type__" in obj and obj["__type__"] == "tuple":
        return tuple(obj["data"])
    return obj

if __name__ == '__main__':
    random.seed()
    random_state = random.getstate()
    
    # 序列化状态到JSON
    json_state = json.dumps(random_state, default=tuple_encoder)
    
    # 从JSON反序列化并恢复状态
    restored_state = json.loads(json_state, object_hook=tuple_decoder)
    random.setstate(restored_state)

方法原理 #

• 序列化时，所有元组会被转为包含__type__标记的字典，JSON可以正常处理该结构。
• 反序列化时，通过object_hook回调识别标记字典，将其转回元组，最终得到和原状态完全一致的嵌套元组结构，满足random.setstate()的类型要求。

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