plt绘画3d散点图且不同类型的点具有不同的颜色举例
时间: 2023-10-10 14:07:39 浏览: 122
好的,以下是一个绘制3D散点图且不同类型的点具有不同颜色的例子:
```python
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
# 生成随机数据
x = [1,2,3,4,5]
y = [6,7,8,9,10]
z = [11,12,13,14,15]
type = [0,1,0,1,0] # 0表示一类点,1表示另一类点
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制不同类型的点
ax.scatter(x=[x[i] for i in range(len(x)) if type[i]==0],
y=[y[i] for i in range(len(y)) if type[i]==0],
zs=[z[i] for i in range(len(z)) if type[i]==0],
c='r', # 一类点红色
marker='o')
ax.scatter(x=[x[i] for i in range(len(x)) if type[i]==1],
y=[y[i] for i in range(len(y)) if type[i]==1],
zs=[z[i] for i in range(len(z)) if type[i]==1],
c='g', # 另一类点绿色
marker='o')
# 设置坐标轴标签
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
plt.show()
```
运行以上代码,就可以绘制出一个3D散点图,不同类型的点具有不同的颜色。其中,红色点表示一类点,绿色点表示另一类点。
阅读全文