风起FQ
2021-09-12 13:54
采纳率: 87.9%
浏览 33
已结题

请问这段python代码为什么总报错?


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()

img

  • 收藏

2条回答 默认 最新

相关推荐 更多相似问题