Python是一種流行的編程語言,適合用于數(shù)據(jù)分析和可視化。Python中有很多庫可以用來畫圖,其中一個流行的庫是matplotlib。在matplotlib中,我們可以使用trendline()函數(shù)來畫趨勢線,并使用get_ydata()函數(shù)來獲取趨勢線上的所有點的y值。
import matplotlib.pyplot as plt # 構造x和y x = [1, 2, 3, 4, 5, 6] y = [2, 5, 7, 8, 9, 11] # 畫圖并畫出趨勢線 plt.plot(x, y, 'o') z = np.polyfit(x, y, 1) # 計算一次多項式擬合線斜率和截距 p = np.poly1d(z) plt.plot(x, p(x), "r--") # 獲取趨勢線上的y值 trendline_y = p(x) print(trendline_y)
上面的代碼中,我們使用polyfit()函數(shù)計算一次多項式擬合線的斜率和截距,然后使用poly1d()函數(shù)創(chuàng)建一個可以計算趨勢線上任意x值對應的y值的多項式函數(shù)。最后,我們使用plot()函數(shù)畫出趨勢線,并使用get_ydata()函數(shù)獲取趨勢線上的所有點的y值。