如何使用polyfit擬合數(shù)據(jù)

小樊
98
2024-06-29 20:49:37
欄目: 編程語言

要使用polyfit來擬合數(shù)據(jù)，首先需要導(dǎo)入numpy庫(kù)，然后使用polyfit函數(shù)進(jìn)行擬合。

以下是一個(gè)簡(jiǎn)單的例子，假設(shè)有一組數(shù)據(jù)x和y，我們想用一個(gè)一次多項(xiàng)式來擬合這組數(shù)據(jù)：

import numpy as np

# 數(shù)據(jù)
x = np.array([1, 2, 3, 4, 5])
y = np.array([2.1, 3.9, 6.1, 8.2, 10.1])

# 用一次多項(xiàng)式擬合數(shù)據(jù)
coefficients = np.polyfit(x, y, 1)
poly = np.poly1d(coefficients)

# 打印擬合的系數(shù)
print("擬合的系數(shù):", coefficients)

# 用擬合的多項(xiàng)式計(jì)算預(yù)測(cè)值
y_fit = poly(x)

# 打印預(yù)測(cè)值
print("預(yù)測(cè)值:", y_fit)

在這個(gè)例子中，使用polyfit函數(shù)擬合了一次多項(xiàng)式，然后使用poly1d函數(shù)創(chuàng)建了一個(gè)多項(xiàng)式對(duì)象poly，并用這個(gè)對(duì)象計(jì)算了預(yù)測(cè)值。

0 踩