plotting.py

import streamlit as st
import pandas as pd
import matplotlib.pyplot as plt

# Plotly imports
from plotly.subplots import make_subplots
import plotly.graph_objects as go
import plotly.figure_factory as ff
import plotly.express as px


def plot(df):
    st.title('Surface grid visualization')
    cmap = plt.cm.jet
    fig, ax = plt.subplots()
    ax.scatter(df.X,df.Y,c=df.Z,cmap=cmap)
    fig.set_size_inches(9, 6)
    st.pyplot(fig)