"""
Scatter Plot Data Slicing
=========================

Example showing data slice for scatter plot.
"""

# test_example = false
# sphinx_gallery_pygfx_docs = 'screenshot'

import fastplotlib as fpl
import numpy as np


figure = fpl.Figure(size=(700, 560))

# create a gaussian cloud of 5_000 points
n_points = 1_000

mean = [0, 0]  # mean of the Gaussian distribution
covariance = [[1, 0], [0, 1]]  # covariance matrix

gaussian_cloud = np.random.multivariate_normal(mean, covariance, n_points)
gaussian_cloud2 = np.random.multivariate_normal(mean, covariance, n_points)

# use an alpha value since this will be a lot of points
scatter1 = figure[0,0].add_scatter(data=gaussian_cloud, sizes=3)
scatter2 = figure[0,0].add_scatter(data=gaussian_cloud2, colors="r", sizes=3)

figure.show()

scatter1.data[:500] = np.array([0 , 0, 0])
scatter2.data[500:] = np.array([0 , 0, 0])

# NOTE: fpl.loop.run() should not be used for interactive sessions
# See the "JupyterLab and IPython" section in the user guide
if __name__ == "__main__":
    print(__doc__)
    fpl.loop.run()