Help with visualization (edgecolors, x-marker with background) #2602
-
|
Hi, I really like this package. edgecolorsAccording to the docs, one should be able to use an attribute "edgecolors" in the agentportrayal. background-colorMy second questions is, is it possible to add an "x" marker in front of an "s" marker, i.e. to set the background of the x marker? |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 4 replies
-
|
Hi! We worked on this in:
Some work has been done on the visualisation in the meantime, so maybe @quaquel knows more about it. Could you give us a Minimal, Reproducible Example? That would help get to the root cause. |
Beta Was this translation helpful? Give feedback.
-
|
|
Beta Was this translation helpful? Give feedback.
-
|
The following is my minimal example. The grid is a MultiGrid (with height/width from model_params). I left out the other agents and components. #Cell is a defined agent
def arsonist_police_portrayal(agent):
if agent is None:
return
portrayal = {
"size": 50,
}
if isinstance(agent, Cell1):
portrayal["color"] = "#ff0000"
portrayal["marker"] = "s"
portrayal["size"] = 300
portrayal["linewidths"] = 3
portrayal["edgecolors"] = "b"
eliif isinstance(agent, Cell2):
portrayal["color"] = "#0ff0"
portrayal["marker"] = "s"
portrayal["size"] = 300
# This needs to be set as well
portrayal["linewidths"] = 3
portrayal["edgecolors"] = "b"
return portrayal
def post_process(ax):
ax.set_aspect("equal")
ax.set_xticks([])
ax.set_yticks([])
ax.get_figure().set_size_inches(10, 10)
space_component = make_space_component(
arsonist_police_portrayal, post_process=post_process, draw_grid=False
)
town_model = TownModel()
model_params = {
"height": 25,
"width": 25
}
page = SolaraViz(
town_model,
components=[space_component],
model_params=model_params,
name="myDashboard",
)The following is the error I get. If I take out edgecolors and linewidths, everything runs without problems. ERROR: Component react.component(mesa.visualization.components.matplotlib_components.SpaceMatplotlib) raised exception IndexError('boolean index did not match indexed array along axis 0; size of axis is 625 but size of corresponding boolean axis is 633') Thanks for the quick responses. |
Beta Was this translation helpful? Give feedback.
-
|
Could you format this code properly? |
Beta Was this translation helpful? Give feedback.
-
|
I am sorry but I don't understand at all what you are doing in agent_portrayal. def arsonist_police_portrayal(agent):
if agent is None:
return
portrayal = {
"size": 50,
}
if isinstance(agent, Cell):
portrayal["color"] = "#ff0000"
portrayal["marker"] = "s"
portrayal["size"] = 300
portrayal["linewidths"] = 3
portrayal["edgecolors"] = "b"
return portrayal
|
Beta Was this translation helpful? Give feedback.
-
|
Hi, I edited my comment. #portrayal["linewidths"] = 3
#portrayal["edgecolors"] = "b"of Cell2 then it works, else you get the error I got. Here is the complete example: Dashboarddef arsonist_police_portrayal(agent):
if agent is None:
return
portrayal = {
"size": 50,
}
if isinstance(agent, Cell1):
portrayal["color"] = "#ff0000"
portrayal["marker"] = "s"
portrayal["size"] = 300
portrayal["linewidths"] = 3
portrayal["edgecolors"] = "b"
elif isinstance(agent, Cell2):
portrayal["color"] = "#ff0000"
portrayal["marker"] = "^"
portrayal["size"] = 300
#portrayal["linewidths"] = 3
#portrayal["edgecolors"] = "b"
portrayal["zorder"] = 2
return portrayal
def post_process(ax):
ax.set_aspect("equal")
ax.set_xticks([])
ax.set_yticks([])
ax.get_figure().set_size_inches(10, 10)
space_component = make_space_component(
arsonist_police_portrayal, post_process=post_process, draw_grid=False
)
town_model = TownModel()
model_params = {
"height": 25,
"width": 25
}
page = SolaraViz(
town_model,
components=[space_component],
model_params=model_params,
name="myDashboard",
)
pageModelfrom mesa import Model
from mesa.space import MultiGrid
from agent import Cell1, Cell2
import numpy as np
class TownModel(Model):
def __init__(self,
width: int = 25,
height: int = 25,
seed=None):
super().__init__(seed=seed)
self.width = width
self.height = height
self.grid = MultiGrid(width, height, torus=False)
self.rng = np.random.default_rng()
for _contents, (x, y) in self.grid.coord_iter():
cell = Cell1((x, y), model=self)
self.grid.place_agent(cell, (x, y))
if x % 5 == 0:
cell2 = Cell2((x, y), model=self)
self.grid.place_agent(cell2, (x, y))Agentsfrom mesa import Agent
class Cell1(Agent):
def __init__(self, pos, model):
super().__init__(model=model)
self.x, self.y = pos
class Cell2(Agent):
def __init__(self, pos, model):
super().__init__(model=model)
self.x, self.y = posSo thanks a lot for your help and sorry for the bad formatting and all. |
Beta Was this translation helpful? Give feedback.
Hi, I edited my comment.
While making a reproducible example I noticed that it ran with edgecolors when only using one agent.
So I tried giving the linewidths and edgecolors parameters to all agents and now it works.
If you remove the comment before
of Cell2 then it works, else you get the error I got.
Here is the complete example:
Dashboard