Plot - Scatter

from __future__ import absolute_import, division, unicode_literals

from future import standard_library
from numpy import linspace, pi, sin

from gemseo.api import configure_logger
from gemseo.core.dataset import Dataset

configure_logger()

standard_library.install_aliases()

Build a dataset

inputs = linspace(0, 1, 20)[:, None]
outputs = sin(2 * pi * inputs)
color = ["b" if abs(output) > 0.5 else "r" for output in outputs]

dataset = Dataset()
dataset.add_variable("x", inputs, "inputs")
dataset.add_variable("y", outputs, "outputs", cache_as_input=False)

Plot y vs x

We can use the Scatter plot

dataset.plot("Scatter", x="x", y="y", color=color)

Out:

<gemseo.post.dataset.scatter.Scatter object at 0x7fc29dc57040>