# Import libraries/dependencies
import numpy as np
import statsmodels.api as sm
from matplotlib import pyplot as plt
from mtbpy import mtbpy

#Statsmodels API reference for QQ plots:
#https://www.statsmodels.org/stable/generated/statsmodels.graphics.gofplots.qqplot.html#statsmodels.graphics.gofplots.qqplot

# Gedding data from Minitab
data_points = mtbpy.mtb_instance().get_column("C1")


# Confirming that data was received
mtbpy.mtb_instance().add_message("C1 data received")

#Converting data from Minitab to NP array
arr = np.array(data_points)

# Generate & save the QQ plot
image = sm.qqplot(arr, line = 'q')
plt.savefig('qqplot.png')

# Send QQ plot image to Minitab
mtbpy.mtb_instance().add_image('qqplot.png')