如何用Python在我的图表上显示R平方值

pip install numpy pandas matplotlib

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

X = np.array([1, 2, 3, 4, 5])
Y = np.array([2, 4, 5, 4, 5])

regression_model = LinearRegression()
regression_model.fit(X.reshape(-1, 1), Y.reshape(-1, 1))
r_squared = regression_model.score(X.reshape(-1, 1), Y.reshape(-1, 1))

plt.scatter(X, Y)
plt.plot(X, regression_model.predict(X.reshape(-1, 1)), color='red')
plt.text(0.5, 0.9, f'R-squared = {r_squared:.2f}', transform=plt.gca().transAxes)
plt.xlabel('X')
plt.ylabel('Y')
plt.title('Linear Regression')
plt.show()