Source code for density_scatter_plot_example

import os
import json
import pandas as pd
from mdvtools.mdvproject import MDVProject
from mdvtools.charts.density_scatter_plot import DensityScatterPlot



def create_scatter_plot(title, param, size, position, color, brush, opacity, radius, legend_display, legend_position, xaxis_properties, yaxis_properties, category1, category2):
    """Create and configure a DensityScatterPlot instance with the given parameters."""
    plot = DensityScatterPlot(
        title=title,
        param=param,
        size=size,
        position=position
    )

    plot.set_default_color(color)
    plot.set_brush(brush)
    plot.set_opacity(opacity)
    plot.set_radius(radius)
    plot.set_color_legend(legend_display, legend_position)
    plot.set_axis_properties("x", xaxis_properties)
    plot.set_axis_properties("y", yaxis_properties)
    plot.set_category1(category1)
    plot.set_category2(category2)
    
    return plot




def load_data(path):
    """Load data from the specified CSV file."""
    return pd.read_csv(path, low_memory=False)




def convert_plot_to_json(plot):
    """Convert plot data to JSON format."""
    return json.loads(json.dumps(plot.plot_data, indent=2).replace("\\\\", ""))




def main():
    """Main function to create the project and serve it."""
    # Constants
    project_path = os.path.expanduser('~/mdv/project')
    data_path = "path_to_data"
    view_name = "default"
    datasource_name = "datasource_name"
    
    # Create project
    project = MDVProject(project_path, delete_existing=True)
    
    # Load data
    data_frame = load_data(data_path)

    data_frame['leiden'] = data_frame['leiden'].astype('category')
 
    # Add datasource
    project.add_datasource(datasource_name, data_frame)
    
    # ScatterPlot parameters
    title = "Density Scatter Plot Example"
    params = ["X_umap_1", "X_umap_2", "leiden"] #param1 and param2 should be a numerical variables, param3 should be categorical variables
    size = [792, 472]
    position = [10, 10]

    color = "#377eb8"
    brush = "poly"
    opacity = 0.8
    radius = 0.2

    legend_display = True
    legend_position = [375,1]
              
    
    xaxis_properties = {"label": "label 1", 
             "textSize": 13, 
             "tickfont": 10
    }

    yaxis_properties = {"label": "label 2", 
             "textSize": 13, 
             "tickfont": 10
    }

    category1 = "1"
    category2 = "2"

    # Create plot
    scatter_plot = create_scatter_plot(title, params, size, position, color, brush, opacity, radius, legend_display, legend_position, xaxis_properties, yaxis_properties, category1, category2)
    
    # Convert plot to JSON and set view
    scatter_plot_json = convert_plot_to_json(scatter_plot)
    scatterplot_view = {'initialCharts': {datasource_name: [scatter_plot_json]}}
    
    project.set_view(view_name, scatterplot_view)
    project.set_editable(True)
    project.serve()


if __name__ == "__main__":
    main()