用matplotlib画一个关于餐厅节假日每天顾客人数和非节假日顾客人数对比的三维柱状图
时间: 2024-02-06 07:10:00 浏览: 23
三维柱状图需要用到mpl_toolkits.mplot3d库,可以按照以下步骤进行绘制。
首先,导入需要的库和数据:
```python
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
# 餐厅每天顾客人数和是否是节假日的数据
data = np.array([
[100, 120, 110, 90, 130, 140, 150],
[150, 170, 160, 140, 180, 190, 200],
[True, False, False, False, True, True, True]
])
```
然后,将数据分成两组,分别是节假日和非节假日:
```python
# 分成两组,分别是节假日和非节假日
holiday_data = data[:, data[2] == True][:2]
non_holiday_data = data[:, data[2] == False][:2]
```
接下来,创建一个3D坐标系,并绘制两组数据的三维柱状图:
```python
# 创建一个3D坐标系
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制节假日数据的柱状图
x = np.arange(len(holiday_data[0]))
y = np.zeros(len(holiday_data[0]))
z = holiday_data[0]
dx = dy = 0.5
dz = holiday_data[1]
ax.bar3d(x, y, z, dx, dy, dz, color='r')
# 绘制非节假日数据的柱状图
x = np.arange(len(non_holiday_data[0]))
y = np.ones(len(non_holiday_data[0]))
z = non_holiday_data[0]
dx = dy = 0.5
dz = non_holiday_data[1]
ax.bar3d(x, y, z, dx, dy, dz, color='b')
# 设置坐标轴标签
ax.set_xlabel('Day')
ax.set_ylabel('Holiday or Not')
ax.set_zlabel('Number of customers')
# 显示图形
plt.show()
```
最终的图形如下所示:
![image.png](attachment:image.png)
可以看到,红色的柱子代表节假日的顾客人数,蓝色的柱子代表非节假日的顾客人数。通过这个三维柱状图,我们可以清晰地看到节假日的顾客人数比非节假日要多。