如何在Python的Mesa库中通过unique_id获取智能体？

算力阿峰

2026-8-16

Mesa中通过Unique ID获取智能体的方法

当然可以，有两种常用的方式实现：

维护ID映射字典（高效推荐）
在你的模型类里新增一个字典，专门用来存储unique_id到智能体实例的映射。这样后续查找智能体的时间复杂度是O(1)，适合大规模模型。
示例代码：

from mesa import Model, Agent
from mesa.time import RandomActivation

class MyModel(Model):
    def __init__(self, num_agents):
        self.schedule = RandomActivation(self)
        # 初始化ID到智能体的映射字典
        self.agent_id_map = {}
        for i in range(num_agents):
            agent = MyAgent(i, self)
            self.schedule.add(agent)
            # 将智能体存入映射字典
            self.agent_id_map[agent.unique_id] = agent

class MyAgent(Agent):
    def __init__(self, unique_id, model):
        super().__init__(unique_id, model)
        # 保存另一个智能体的ID（示例）
        self.target_agent_id = 5

    def step(self):
        # 通过ID获取目标智能体
        target_agent = self.model.agent_id_map.get(self.target_agent_id)
        if target_agent:
            # 执行交互逻辑，比如发送消息、修改状态等
            print(f"Agent {self.unique_id} interacting with Agent {target_agent.unique_id}")

遍历调度器的智能体列表（适合小规模模型）
如果模型里智能体数量不多，也可以直接遍历模型调度器中的所有智能体，匹配对应的unique_id。这种方法不需要额外维护字典，但查找效率是O(n)，智能体多的时候会变慢。
示例代码：
```
def get_agent_by_id(model, target_id):
    for agent in model.schedule.agents:
        if agent.unique_id == target_id:
            return agent
    # 没找到返回None
    return None

# 在智能体的step方法中调用
target_agent = get_agent_by_id(self.model, self.target_agent_id)
```