Python Function Change Input Variable Matlab Plot Colors