-
Integers, Floats and Strings
-
If Statements
-
Lists, Sets and Tuples
-
Loops
-
Functions
-
Dictionaries
-
Assertions
-
Classes
-
Matplotlib
Plotting Part 2
Adding Color¶
Colors may be used for lines by using the argument color with the plot function. There are many colors to choose from, which the matplotlib documentation can give you ideas about. For now, we will use r and g which give us the red and green colors. Below we modify the code so that we set the colors.
In [9]:
#Plot the cost with a label and red color
plt.plot(volume, cost, label="Cost", color='r')
#Plot the revenue with a label and green color
plt.plot(volume, revenue, label="Revenue", color='g')
#Add an x label
plt.xlabel("Units")
#Add a y label
plt.ylabel("$")
#Add a title
plt.title("Total Cost Function")
#Call for the legend
plt.legend()
#Show the plot
plt.show()
Add in a new series of data which is the profit function.
In [10]:
#Find the profit
profit = [a-b for a,b in zip(revenue, cost)]
print(profit)
In [11]:
#Plot the cost with a label and red color
plt.plot(volume, cost, label="Cost", color='r')
#Plot the revenue with a label and green color
plt.plot(volume, revenue, label="Revenue", color='g')
#Plot the revenue with a label and green color
plt.plot(volume, profit, label="Profit", color='black')
#Add an x label
plt.xlabel("Units")
#Add a y label
plt.ylabel("$")
#Add a title
plt.title("Total Cost Function")
#Call for the legend
plt.legend()
#Show the plot
plt.show()
Playing with the linestyles will let us switch out from the straight lines to some other types. Examples are below.
In [12]:
#Plot the cost with a label and red color
plt.plot(volume, cost, label="Cost", color='r', linestyle="--")
#Plot the revenue with a label and green color
plt.plot(volume, revenue, label="Revenue", color='g',linestyle="dotted")
#Plot the revenue with a label and green color
plt.plot(volume, profit, label="Profit", color='black')
#Add an x label
plt.xlabel("Units")
#Add a y label
plt.ylabel("$")
#Add a title
plt.title("Total Cost Function")
#Call for the legend
plt.legend()
#Show the plot
plt.show()
Prev
Plotting