Python Use Function Output Matlab Colors