Note that by including the return character, backslash n, we are creating a two-line title in this character string. In the right-hand plot, this default title has been overridden by specifying the optional argument main. The plot on the left uses the default title returned by the density function, which tells us that the plot was generated by this function using its default options and also gives the R specification for the variable whose density we are plotting. Plot(density(geyser$waiting), main = 'Estimated density: Old Faithful waiting times') Some base graphics functions also provide default titles, but again, these can be overridden has we see here: library(MASS) Base graphics functions in R typically create axis labels by default, although these can be overridden through the argument xlab() that allows us to provide our own x-axis label and ylab() that allows us to provide our own y-axis label. I really wanted the first solution to work where I just had a custom dictionary and looked it up during the call to axes.scatter, but the errors were just bizarre and not understandable.Plot character or pch is the standard argument to set the character that will be plotted in a number of R functions.Įxplanatory text can be added to a plot in several different forms, including axis labels, titles, legends, or a text added to the plot itself. I tried to use the np.where statement, but I have 3 options for sex (Male, Female, and Infant, abbreviated as M,F, I) data = np.where(data.Sex = 'M', 'Firebrick', (data.Sex = 'F', 'blueviolet','beige'))Īnd finally I got this to work: def label_color(row):ĭata = data.apply(lambda row: label_color(row), axis=1)īut I wasn't all that satisfied with the solution. I tried to make a dictionary from scratch that had the values in them: weight_dict = pd.DataFrame(dict(whole = data, shucked = data, sex = data, color= some if statement that choked)) Next I tried to loop through the df and add the column manually based on a row value: for x1 in data: First I tried to index a dictionary based on the df column called 'Sex' figure = plt.figure(figsize=(20, 6))Ĭlr = Īxes.scatter( data, data,color=clr)])Īxes.set_title('Whole Weight vs.

I've tried a bunch of different ways to get this to work. I have a pandas dataframe that has data I want to plot, but I'd like to change the color of the dot based on the sex of the animal.

Multi scatter character