python的meshgrid用法和3D庫 mpl_toolkits.mplot3d 與PolynomialFeatures多項式庫學習

meshgrid

 

import numpy as np
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import  Axes3D
x = np.array([0,1,2])
y = np.array([0,1])
X,Y = np.meshgrid(x,y)#X,Y擴展成了矩陣，
print(X)
print(Y)
theta0, theta1, theta2 = 2, 3, 4
ax = Axes3D(plt.figure())#用來畫三維圖
Z = theta0 + theta1*X + theta2*Y#求z值
plt.plot(X,Y,'r.')#此時你會發現繪畫出的是3*2個點，這些點組成一個網格，切每個點的坐標是X*Y的笛卡爾積
ax.plot_surface(X,Y,Z)#用來畫三維圖
plt.show()

具體也可以參考這篇博客

mpl_toolkits.mplot3d

關於3D繪圖的博客

繪圖時用到划分面板的方法

使用add_subplot

 

import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 100)
fig = plt.figure()
ax1 = fig.add_subplot(221)
ax1.plot(x, x)
ax2 = fig.add_subplot(222)
ax2.plot(x, -x)
ax3 = fig.add_subplot(223)
ax3.plot(x, x ** 2)
ax4 = fig.add_subplot(224)
ax4.plot(x, np.log(x))
plt.show()

 

使用subplot方法

import numpy as np
from matplotlib import pyplot as plt
x = np.arange(10)
plt.subplot(221)
plt.plot(x,x)
plt.subplot(223)
plt.plot(x,-x)
plt.show()

 

 

 

PolynomialFeatures

import numpy as np
import matplotlib.pyplot as plt
from sklearn.preprocessing import PolynomialFeatures#多項式
from sklearn.linear_model import LinearRegression

# 載入數據
data = np.genfromtxt("job.csv", delimiter=",")
x_data = data[1:,1]
y_data = data[1:,2]
plt.scatter(x_data,y_data)
plt.show()
#維度必須是二維
x_data = x_data[:,np.newaxis]
y_data = y_data[:,np.newaxis]
# 定義多項式回歸,degree的值可以調節多項式的特征
poly = PolynomialFeatures(degree=4)
# 特征處理
x_poly = poly.fit_transform(x_data)
# 定義回歸模型
model = LinearRegression()
# 訓練模型
model.fit(x_poly,y_data)
plt.plot(x_data,y_data,'b.')
plt.plot(x_data,model.predict(poly.fit_transform(x_data)),'r')
plt.show()