import numpy as np
import pandas as pd

df = pd.read_csv('salinity_salinitychange.csv')

least_salty = df[df['Salinity'] == 'Least Salty']
medium_salty = df[df['Salinity'] == 'Medium Salty']
most_salty = df[df['Salinity'] == 'Most Salty']

medium_salty['Input Frequency [Hz]'] = least_salty['Input Frequency [Hz]'].to_numpy()
most_salty['Input Frequency [Hz]'] = least_salty['Input Frequency [Hz]'].to_numpy()

fig, axis = plt.subplots()

least_salty.plot.scatter(x = 'Input Frequency [Hz]', y = 'Output Amplitude [Vpp]', ax = axis, label = 'Least Salty', c = "red")
medium_salty.plot.scatter(x = 'Input Frequency [Hz]', y = 'Output Amplitude [Vpp]', ax = axis, label = 'Meduim Salty', c = "blue")
most_salty.plot.scatter(x = 'Input Frequency [Hz]', y = 'Output Amplitude [Vpp]', ax = axis, label = 'Most Salty', c = "green")

plt.show()