Создание штабелированного участка цилиндра в matplotlib

Question

Может ли matplotlib нарисовать штабелированный цилиндр, как показано ниже? Как насчет того, имеет ли он только один бар?

Если нет, другой вариант был бы просто обычным штабелем с барами с закругленными краями - возможно ли это?

Я искал в галерее matplotlib и bar() документации, но не смог найти что-то для этого.

Answer 1

Я не знаю, прямой функции для построения многослойную цилиндра бар участок, и я не думаю, что есть простой обходной путь либо. Проблема в том, что он не является ни 2D, ни 3D.

С помощью Matplotlib вы должны сделать 2D-изображение 3D. Это означает, что вы должны сделать форму цилиндра. Чтобы он выглядел хорошо, вам, вероятно, понадобится текстура, чтобы придать внешний вид тени.

mplot3d является трехмерным расширением Matplotlib, и я использовал его для создания графика ниже. Я думаю, что это выглядит немного 3D. Верх выглядит немного искаженным, и весь сюжет находится под углом ... mplot3d тоже немного больно работать. Требуется немало усилий, чтобы цилиндр выглядел красиво. Код не очень полированный, но я его аннотировал.

from __future__ import print_function 
from __future__ import division 
from __future__ import absolute_import 

from mpl_toolkits.mplot3d import Axes3D 

import numpy 
import matplotlib 
import matplotlib.pyplot as plt 


def plot_cylinder_element(x, z, dz, rx = 5, ry = 5, color = "b"): 
    """ 
    x: left, right 
    z: start height 
    dz: height of cylinder 
    rx, ry = radius of width (x) and depth (y) 
    color = color 

    Inspired by: 
http://matplotlib.1069221.n5.nabble.com/plot-surface-shading-and-clipping-error-td14031.html 
    """ 

    N = 100    # number of elements 
    # a lower stride will give more faces. A cylinder with 4 faces is a cube :) 
    # I think with N=100 and rstride=2, it will have 50 faces 
    # cstride is the height, rstride the circle 
    cstride_side = 1000 # only 1 element needed 
    rstride_side = 1 # many elements to make a nice cylinder shape 
    cstride_top = 10  
    rstride_top = 10 

    # parameters of cylinder 
    phi = numpy.linspace(0, 2 * numpy.pi, N) 
    _r = numpy.ones(N) 
    _h = numpy.linspace(0, 1, N) 

    # cylinder 
    _x = rx * numpy.outer(numpy.cos(phi), _r) + x 
    _y = ry * numpy.outer(numpy.sin(phi), _r) 
    _z = dz * numpy.outer(numpy.ones(numpy.size(_r)), _h) + z 
    ax.plot_surface(_x, _y, _z, rstride = rstride_side, cstride = cstride_side, linewidth = 0, alpha = 1, color = color) 

    # to cover the gaps between the faces, plot the cylinder again at a slightly smaller radius 
    _x *= 0.99 
    _y *= 0.99 
    ax.plot_surface(_x, _y, _z, rstride = rstride_side + 1, cstride = cstride_side + 1, linewidth=0, alpha=1, color = color) 

    # top 
    _x = rx * numpy.outer(numpy.cos(phi), _h) + x 
    _y = ry * numpy.outer(numpy.sin(phi), _h) 
    _z = numpy.zeros([N,N]) + z + dz + 0.1 
    ax.plot_surface(_x, _y, _z, rstride = rstride_top, cstride = cstride_top, linewidth = 0, alpha = 1, color = color) 

    # plot again with different stride to mask the gaps  
    ax.plot_surface(_x, _y, _z, rstride = rstride_side + 1, cstride = cstride_side + 1, linewidth=0, alpha=1, color = color) 


def plot_cylinder(x, z, rx = 5, ry = 5): 
    """ 
    x: left-right for each cylinder 
    z: list height difference (ie. not cumulative) 
    """ 
    # list with colors 
    colors = ["b", "g", "r", "c", "y", "k"] 
    # plot cylinder elements 
    _z = 0 
    for i in range(len(z)): 
     plot_cylinder_element(x, _z, z[i], rx = rx, ry = ry, color = colors[i % len(colors)]) 
     _z += z[i] 


def cylinder_plot(z, r = 10, dr = 30): 
    """ 
    z: list of different cylinders with for each a list height difference (ie. not cumulative) 
    r: radius 
    dr: distance between cylinders  
    """ 
    # different cylinders next to each other 
    x = numpy.arange(len(z)) * dr 
    # possible difference between width (x) and depth (y) 
    rx = r 
    ry = r 
    # make cylinders 
    for i in range(len(z)): 
     plot_cylinder(x[i], z[i], rx = rx, ry = ry) 


# close earlier plots 
plt.close("all") 

# make figure 
fig = plt.figure() 
ax = Axes3D(fig) 

# set 3D-view 
ax.view_init(elev = 10, azim = 280) 

# make 3 cylinders, with a different number of elements 
cylinder_plot([[5, 10, 5], [3, 5], [1,2,3,4]]) 

# set the labels 
ax.set_xlabel('X') 
ax.set_ylabel('Y') 
ax.set_zlabel('Z') 

# show 
plt.show()