Python Use Function Output Matlab Plot Types