import matplotlib.pyplot as plt
import numpy as np
import csv
from mpl_toolkits.mplot3d import Axes3D
#定义坐标轴
fig = plt.figure()
ax4 = plt.axes(projection='3d')
#ax = fig.add_subplot(111,projection='3d') #这种方法也可以画多个子图
filename="D:/data/fujian1.csv"
with open(filename) as f:
reader=csv.reader(f)
header_row=next(reader)
X,Y,Z=[ ],[ ],[ ]
for row in reader:
X=int(float(row[1]))
Xs = np.array(X)
Y = int(float(row[2]))
Ys = np.array(Y)
Z= int(float(row[3]))
Zs = np.array(Z)
#作图
ax4.plot_surface(Xs,Ys,Zs,alpha=0.3,cmap='winter') #生成表面, alpha 用于控制透明度
ax4.contour(Xs,Ys,Zs,zdir='Zs', offset=-3,cmap="rainbow") #生成z方向投影,投到x-y平面
ax4.contour(Xs,Ys,Zs,zdir='Xs', offset=-6,cmap="rainbow") #生成x方向投影,投到y-z平面
ax4.contour(Xs,Ys,Zs,zdir='Ys', offset=6,cmap="rainbow") #生成y方向投影,投到x-z平面
#ax4.contourf(Xs,Ys,Zs,zdir='y', offset=6,cmap="rainbow") #生成y方向投影填充,投到x-z平面,contourf()函数
ax4.set_xlabel('X')
ax4.set_xlim(-300, 300) #拉开坐标轴范围显示投影
ax4.set_ylabel('Y')
ax4.set_ylim(-300, 300)
ax4.set_zlabel('Z')
ax4.set_zlim(-300, 300)
plt.savefig('figure2.png',dpi=500)
plt.show()
同学你好,散点是无法生成surface的,必须是网格点2D才能生成平面
还有你这句
for row in reader:
X=int(float(row[1]))
Xs = np.array(X)
Y = int(float(row[2]))
Ys = np.array(Y)
Z= int(float(row[3]))
Zs = np.array(Z)
要改成
for row in reader:
X.append(int(float(row[1])))
Xs = np.array(X)
Y.append(int(float(row[2])))
Ys = np.array(Y)
Z.append(int(float(row[3])))
Zs = np.array(Z)