python-matplotlibHow to plot bestfit curve line
In order to make curved line, we have to increase degree of our polynomial (3rd argument to polyfit()):
import matplotlib.pyplot as plt
import numpy as np
x = np.array([1, 3, 5, 7])
y = np.array([6, 6, 7, 8])
plt.plot(x, y, 'o')
a, b, c = np.polyfit(x, y, 2)
plt.plot(x, a * x*x + b*x + c)
plt.show()ctrl + cimport matplotlib.pyplot as pltloads Matplotlib module to use plotting capabilities | [1, 3, 5, 7]list of x coordinates of dots to plot best fit regression from |
[6, 6, 7, 8]list of y coordinates of dots to plot best fit regression from | polyfitcalculates least square polynomial fit |
a * x*x + b*x + cwe plot 2-degree polynomial to get curved best fit regression line | .show()render chart in a separate window |
Usage example
import matplotlib.pyplot as plt
import numpy as np
x = np.array([1, 3, 5, 7])
y = np.array([6, 6, 7, 8])
plt.plot(x, y, 'o')
a, b, c = np.polyfit(x, y, 2)
plt.plot(x, a * x*x + b*x + c)
plt.show()Related
More of Python Matplotlib
- Errorbar usage example
- How to set more space between subplots
- How to add legend to boxplot
- How to plot multiple boxplots
- How to plot heatmap with values
- How to plot multiple charts
- How to change boxplot colors
- How to install GeoPandas module
- How to export to pdf
- How to draw line chart
See more codes...