python 画3d条形图,根据z轴坐标改变颜色
时间: 2024-12-27 20:32:41 浏览: 13
在Python中,你可以使用matplotlib库中的mplot3d模块来创建3D条形图,并通过设置`color`参数来根据Z轴的值改变每个条形的颜色。下面是一个简单的例子:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建数据
x = np.array([0, 1, 2, 3])
y = np.array([4, 5, 6, 7])
z = np.random.rand(4) # 随机生成Z轴值
# 创建3D条形图
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 根据Z轴值给条形赋予颜色
for i in range(len(x)):
color = plt.cm.viridis(z[i]) # 使用viridis颜色映射,可以根据需要替换为其他颜色映射
ax.bar3d(x[i], y[i], z[i], 1, 1, z[i]-min(z), color=color)
ax.set_xlabel('X Axis')
ax.set_ylabel('Y Axis')
ax.set_zlabel('Z Axis')
plt.show()
阅读全文