646134029 发表于 2021-12-18 16:41:07

matplotlib求助

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.), *zip(x, y), (x[-1], 0.)]


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

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

# verts is a list of (x, y) pairs defining polygon i.
verts =
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
有没有大佬帮忙处理一下,搞不定

646134029 发表于 2021-12-18 16:42:39

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

646134029 发表于 2021-12-18 16:54:36

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

suchocolate 发表于 2021-12-18 18:00:10

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

疾风怪盗 发表于 2021-12-18 18:11:36

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

阿奇_o 发表于 2021-12-18 18:22:03

本帖最后由 阿奇_o 于 2021-12-18 18:24 编辑

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会改变,从而导致一些函数用不了或不是旧代码所需要的返回。
页: [1]
查看完整版本: matplotlib求助