Three dimensional scatter plot matplotlib legend