python绘图+ML

1. 散点图

# 这种方式好像不太符合
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

#构建xyz
X = np.array([1.0, 2.0, 3.0, 6.0])
Y = np.array([2.0, 4.0, 6.0, 8.0])
Z = np.array([5.0, 10.0, 15.0, 20.0])

fig = plt.figure()
ax = Axes3D(fig)
ax.scatter(X,Y,Z,c=np.sqrt(X ** 2 + Y ** 2 + Z ** 2),cmap="jet", marker="o")
plt.show()






# 给定x,y,z绘制散点图
# import matplotlib.pyplot as plt
# from mpl_toolkits.mplot3d import Axes3D
#
# fig = plt.figure()
# ax = fig.add_subplot(projection='3d')
#
# x = [1.0, 2.0, 3.0, 6.0]
# y = [2.0, 4.0, 6.0, 8.0]
# z = [5.0, 10.0, 15.0, 20.0]

# ax.scatter(x, y, z)
#
# plt.show()

2. 二元一次方程拟合

import numpy as np
from sklearn.linear_model import LinearRegression

#输入数据
X = np.array([[1, 2], [2, 4], [3, 6], [4, 8]])
Y = np.array([5, 10, 15, 20])

#创建模型
model = LinearRegression()

#拟合模型
model.fit(X, Y)

print("系数: ", model.coef_)
print("截距: ", model.intercept_)