ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

python – 使用Matplotlib的3D绘图:隐藏轴但保留轴标签?

2019-10-04 11:56:33  阅读:280  来源: 互联网

标签:python matplotlib plot mplot3d


我正在使用Matplotlib来可视化三维数组.
我几乎按照我想要的方式得到了它,除了一个小小的障碍…请参阅下面的插图和描述我可以做什么以及我想要它做什么…

>展示一堆带有标签的立方体,还有一堆其他的东西.
>显示一堆立方体但没有轴标签.
>这就是我想要但却无法做到的事情……我想展示一堆带有轴标签的立方体,但没有别的.

我希望你们能帮助我:)请参阅下面的来源.

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
from matplotlib import rcParams
import numpy as np
rcParams['axes.labelsize'] = 14
rcParams['axes.titlesize'] = 16
rcParams['xtick.labelsize'] = 14
rcParams['ytick.labelsize'] = 14
rcParams['legend.fontsize'] = 14
rcParams['font.family'] = 'serif'
rcParams['font.serif'] = ['Computer Modern Roman']
rcParams['text.usetex'] = True
rcParams['grid.alpha'] = 0.0

def make_cube():
    """ A Cube consists of a bunch of planes..."""

    planes = {
        "top"    : ( [[0,1],[0,1]], [[0,0],[1,1]], [[1,1],[1,1]] ),
        "bottom" : ( [[0,1],[0,1]], [[0,0],[1,1]], [[0,0],[0,0]] ),
        "left"   : ( [[0,0],[0,0]], [[0,1],[0,1]], [[0,0],[1,1]] ),
        "right"  : ( [[1,1],[1,1]], [[0,1],[0,1]], [[0,0],[1,1]] ),
        "front"  : ( [[0,1],[0,1]], [[0,0],[0,0]], [[0,0],[1,1]] ),
        "back"   : ( [[0,1],[0,1]], [[1,1],[1,1]], [[0,0],[1,1]] )
    }
    return planes

def render_array(ary, highlight):
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')

    cube = make_cube()

    for space in xrange(0, ary.shape[0]):
        for column in xrange(0, ary.shape[1]):
            for row in xrange(0, ary.shape[2]):
                alpha = 0.01
                if highlight[space,column,row] == 1:
                    alpha = 1
                for side in cube:
                    (Xs, Ys, Zs) = (
                        np.asarray(cube[side][0])+space+space*0.2,
                        np.asarray(cube[side][2])+row+row*0.2,
                        np.asarray(cube[side][3])+column+column*0.2
                    )
                    ax.plot_surface(Xs, Ys, Zs, rstride=1, cstride=1, alpha=alpha)

    highest = 0                         # Make it look cubic
    for size in ary.shape:
        if size > highest:
            highest = size
    ax.set_xlim((0,highest))
    ax.set_ylim((0,highest))
    ax.set_zlim((0,highest))


    ax.set_xlabel('Third dimension' )   # Meant to visualize ROW-MAJOR ordering 
    ax.set_ylabel('Row(s)')
    ax.set_zlabel('Column(s)')

    #plt.axis('off')    # This also removes the axis labels... i want those...
    #ax.set_axis_off()  # this removes too much (also the labels)

    # So I try this instead...
    ax.set_xticks([])          # removes the ticks... great now the rest of it
    ax.set_yticks([])
    ax.set_zticks([])
    #ax.grid(False)             # this does nothing....
    #ax.set_frame_on(False)     # this does nothing....
    plt.show()

def main():

    subject = np.ones((3,4,3))

    highlight = np.zeros(subject.shape) # Highlight a row
    highlight[1,1,:] = 1

    render_array(subject, highlight)    # Show it

if __name__ == "__main__":
    main()

更新,感谢答案,这是我缺少的:

# Get rid of the panes                          
ax.w_xaxis.set_pane_color((1.0, 1.0, 1.0, 0.0)) 
ax.w_yaxis.set_pane_color((1.0, 1.0, 1.0, 0.0)) 
ax.w_zaxis.set_pane_color((1.0, 1.0, 1.0, 0.0)) 

# Get rid of the spines                         
ax.w_xaxis.line.set_color((1.0, 1.0, 1.0, 0.0)) 
ax.w_yaxis.line.set_color((1.0, 1.0, 1.0, 0.0)) 
ax.w_zaxis.line.set_color((1.0, 1.0, 1.0, 0.0))

哪个,以及:

# Get rid of the ticks                          
ax.set_xticks([])                               
ax.set_yticks([])                               
ax.set_zticks([])

隐藏除标签之外的所有内容,如3)所示.

UPDATE

我已经清理并将代码置于工作状态并在此处可用:
https://github.com/safl/ndarray_plot

除了这里的几个附加示例:
http://nbviewer.ipython.org/github/safl/ndarray_plot/blob/master/nb/ndap.ipynb

解决方法:

我试图运行你的代码,但它不能在我的电脑上运行.

您可以在下面看到绘制球体的解决方案.基本上,我手动将窗格和刺的颜色变为alpha = 0,并使得刻度为空列表(正如您所指出的).

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
from matplotlib import rcParams
import numpy as np
rcParams['axes.labelsize'] = 18
rcParams['font.family'] = 'serif'
rcParams['font.serif'] = ['Computer Modern Roman']
rcParams['text.usetex'] = True

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

u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(0, np.pi, 100)

x = 10 * np.outer(np.cos(u), np.sin(v))
y = 10 * np.outer(np.sin(u), np.sin(v))
z = 10 * np.outer(np.ones(np.size(u)), np.cos(v))
ax.plot_surface(x, y, z,  rstride=4, cstride=4, color='b')

# Get rid of the panes
ax.w_xaxis.set_pane_color((1.0, 1.0, 1.0, 0.0))
ax.w_yaxis.set_pane_color((1.0, 1.0, 1.0, 0.0))
ax.w_zaxis.set_pane_color((1.0, 1.0, 1.0, 0.0))

# Get rid of the spines
ax.w_xaxis.line.set_color((1.0, 1.0, 1.0, 0.0))
ax.w_yaxis.line.set_color((1.0, 1.0, 1.0, 0.0))
ax.w_zaxis.line.set_color((1.0, 1.0, 1.0, 0.0))

# Get rid of the ticks
ax.set_xticks([]) 
ax.set_yticks([]) 
ax.set_zticks([])

# Add the labels
ax.set_xlabel('Third dimension' )
ax.set_ylabel('Row(s)')
ax.set_zlabel('Column(s)')
plt.show()

这是我的输出

标签:python,matplotlib,plot,mplot3d
来源: https://codeday.me/bug/20191004/1852786.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有