Python ApacheZeppelin在使用matplotlib时出错

当我在pycharm中运行脚本时，下面的代码工作正常

import pandas as pd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.patches as mpatches

c1 = mpatches.Patch(color="green",label="No Traffic")

c2 = mpatches.Patch(color="red",label="Traffic")

df = predict_df.limit(100).toPandas()

colors = {0:"red",1:"green"}


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

ax.scatter(df["avgSpeed"],df["vehicleCount"],df["avgMeasuredTime"],c=df["prediction"].apply(lambda x: colors[x]),s=100,marker="o")

ax.set_xlabel('avgMeasuredTime')

ax.set_ylabel('avgSpeed')

ax.set_zlabel('vehicleCount')


plt.title("BiKmeans Traffic Data")


plt.legend(handles=[c1,c2])
plt.show()

我已经使用pip安装了matplotlib，这也是基于一些类似的问题，试图安装为sudo apt get install python matplotlib 但我在Zepplin也有同样的错误

 File "/home/benjamin/.local/lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.py", line 280, in <lambda>
    key=lambda col: col.do_3d_projection(renderer), AttributeError: 'PathCollection' object has no attribute 'do_3d_projection'

文件“/home/benjamin/.local/lib/python2.7/site packages/mpl_toolkit/mplot3d/axes3d.py”，第280行，在
key=lambda col:col.do_3d_投影（渲染器），AttributeError:“PathCollection”对象没有属性“do_3d_投影”

matplotlib的版本是2.2.0