[已解决]matplotlib求助

646134029

from matplotlib.collections import PolyCollection
import matplotlib.pyplot as plt
import numpy as np
from scipy.stats import poisson

# Fixing random state for reproducibility
np.random.seed(19680801)


def polygon_under_graph(x, y):
    """
    Construct the vertex list which defines the polygon filling the space under
    the (x, y) line graph. This assumes x is in ascending order.
    """
    return [(x[0], 0.), *zip(x, y), (x[-1], 0.)]


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

x = np.linspace(0., 10., 31)
lambdas = range(1, 9)

# verts[i] is a list of (x, y) pairs defining polygon i.
verts = [polygon_under_graph(x, poisson.pmf(l, x)) for l in lambdas]
facecolors = plt.colormaps['viridis_r'](np.linspace(0, 1, len(verts)))

poly = PolyCollection(verts, facecolors=facecolors, alpha=.7)
ax.add_collection3d(poly, zs=lambdas, zdir='y')

ax.set(xlim=(0, 10), ylim=(1, 9), zlim=(0, 0.35),
       xlabel='x', ylabel=r'$\lambda$', zlabel='probability')

plt.show()

这是我在matplotlib官方网站看到的一个绘图案例，但是这官方的例子居然报错？
链接：https://matplotlib.org/stable/gallery/mplot3d/polys3d.html
有没有大佬帮忙处理一下，搞不定

最佳答案

月排行榜 / 总排行榜

阿奇_o

2021-12-18 18:22:03

646134029 发表于 2021-12-18 16:54
我试过其实把报错那一行删掉就可以正常跑了，但是颜色都成一样的了，我目前不太知道怎么设置每个多边形的颜 ...

你多搜几下相关的API和例子，并看看关键的几个函数返回什么，参数facecolors需要怎样的数据，
就会发现，改为 cm.get_cmap() 就可以了，如下

from matplotlib import cm
facecolors = cm.get_cmap('viridis_r', len(verts))(np.linspace(0, 1, len(verts)))  # 
print(facecolors)

ps：这种通常是版本问题，一些新版本的API会改变，从而导致一些函数用不了或不是旧代码所需要的返回。

跳转到最佳答案楼层

646134029

救救孩子，绘图模块真是难学啊，我弄了几个小时还是搞不定

646134029

我试过其实把报错那一行删掉就可以正常跑了，但是颜色都成一样的了，我目前不太知道怎么设置每个多边形的颜色，如果能分色的话可以不用修改源代码，我只需要每个多边形填充一样的颜色
求帮助

suchocolate

报错信息发一下啊，我这运行颜色都正常。

疾风怪盗

下载了运行没有报错，颜色也一样，没问题

阿奇_o

646134029 发表于 2021-12-18 16:54
我试过其实把报错那一行删掉就可以正常跑了，但是颜色都成一样的了，我目前不太知道怎么设置每个多边形的颜 ...

你多搜几下相关的API和例子，并看看关键的几个函数返回什么，参数facecolors需要怎样的数据，
就会发现，改为 cm.get_cmap() 就可以了，如下

from matplotlib import cm
facecolors = cm.get_cmap('viridis_r', len(verts))(np.linspace(0, 1, len(verts)))  # 
print(facecolors)

ps：这种通常是版本问题，一些新版本的API会改变，从而导致一些函数用不了或不是旧代码所需要的返回。