Python Use Function Output Matlab Plot