使用matplotlib在python中绘制散点图

2024-04-20 07:22:43 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个有3000点的档案。我得用这些画一个三维散点图点。点要点如下:

6.655597594660206395e-01,-5.205175889492101859e-01,4.583497554501108073e-01
3.357418723194116605e-02,-2.482341476533565849e+00,2.009030294705723030e+00
-1.398411818716352728e-01,-1.990250936356241063e+00,2.325394845551588929e+00

有3000个这样的点。你知道吗

我有代码,它显示了错误

TypeError: Cannot cast array data from dtype('float64') to dtype('<U32') according to the rule 'safe'

我写的代码

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

x=[]
y=[]
z=[]

with open('3dpd.out') as f:
    lines = f.readlines()
    for line in lines :
        x.append(line.split(',')[0])
        y.append(line.split(',')[1])
        z.append(line.split(',')[2])


fig = plt.figure()
ax = fig.add_subplot(111,projection='3d')

ax.set_title("Scatter Plot")
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')


ax.scatter(x, y, z, c='r')

plt.show()

Tags: to代码fromimportaslinefigplt
1条回答
网友
1楼 · 发布于 2024-04-20 07:22:43

当行被readlines读取时,它们作为字符串读入,因此需要将这些字符串转换为数字。你知道吗

另外,请注意,我在代码中将每个步骤分隔成不同的行(例如,splitfloat,等等),这通常有助于调试。(每行只需调用split一次。)

下面是一个有效的例子:

enter image description here

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

x=[]
y=[]
z=[]

with open('3dpd.out') as f:
    lines = f.readlines()
    for line in lines :
        xs, ys, zs =  line.split(',')
        xv, yv, zv = [float(v) for v in (xs, ys, zs)]
        x.append(xv)
        y.append(yv)
        z.append(zv)

print type(xs)
print type(xv)

fig = plt.figure()
ax = fig.add_subplot(111,projection='3d')

ax.set_title("Scatter Plot")
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')


ax.scatter(x, y, z, c='r')

plt.show()

相关问题 更多 >