code_snippets/reference_designs/scripts/plot_sal.py

20 lines
838 B
Python

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()