taipy/doc/gui/examples/charts/scatter-styling.py

# Copyright 2021-2024 Avaiga Private Limited
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
# an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
# specific language governing permissions and limitations under the License.
# -----------------------------------------------------------------------------------------
# To execute this script, make sure that the taipy-gui package is installed in your
# Python environment and run:
#     python <script>
# You may need to install the scikit-learn package as well.
# -----------------------------------------------------------------------------------------
import numpy as np
import pandas as pd
from sklearn.datasets import make_classification

from taipy.gui import Gui

if __name__ == "__main__":
    # Let scikit-learn generate a random 2-class classification problem
    n_samples = 100
    features, label = make_classification(n_samples=n_samples, n_features=2, n_informative=2, n_redundant=0)

    random_data = pd.DataFrame({"x": features[:, 0], "y": features[:, 1], "label": label})

    data_x = random_data["x"]
    class_A = [random_data.loc[i, "y"] if random_data.loc[i, "label"] == 0 else np.nan for i in range(n_samples)]
    class_B = [random_data.loc[i, "y"] if random_data.loc[i, "label"] == 1 else np.nan for i in range(n_samples)]

    data = {"x": random_data["x"], "Class A": class_A, "Class B": class_B}
    marker_A = {"symbol": "circle-open", "size": 16}
    marker_B = {"symbol": "triangle-up-dot", "size": 20, "opacity": 0.7}
    page = """
# Scatter - Styling

<|{data}|chart|mode=markers|x=x|y[1]=Class A|y[2]=Class B|width=60%|marker[1]={marker_A}|marker[2]={marker_B}|>
    """

    Gui(page).run()