我如何在 Python 中使用 matplotlib plot 一条最佳拟合线？

Question

I'm trying to plot a line of best fit for this graph. I'm having trouble since x is a categorical variable, and all the solutions I've found don't work if one of them is a categorical variable.

Here's my code for the plot -

plt.title('Timeline of cases in Maharashtra')

plt.scatter(maharashtra_confirmed['Date'], maharashtra_confirmed['Maharashtra'], label = 'Maharashtra', color = 'orange')
plt.xticks(rotation = 90)
ax = plt.axes()
ax.xaxis.set_major_locator(ticker.MultipleLocator(2))
plt.legend()
plt.show()

Answer 1

For categorical variables, a fit would not make sense from a theoretical view point. But what you have here is not per-se categorical data, but (as far as I can guess from your input) just data that can not automatically be interpreted as non-categorical.

So you have to convert the dates into a more adequate representation. datetime.datetime.strptime should help you here.