Python Axes3D自制坐標軸

???因為很不巧用Axes3D自帶的坐標軸畫出來圖有點丑，首先ticklabels離坐標軸有點遠。其次想正負值都能顯示。就在想能不能把軸往上移一下，同時拉近ticklabels和坐標軸的距離。

???博主主要是發(fā)現(xiàn)了用ax.axis(‘off’)或者ax1.set_axis_off()可以達到把坐標軸全部關掉的效果，如下：

???那么使用ax.plot畫出三條線（顏色、粗細均可調(diào)整），再配合上ax.text在對應的位置寫上ticklabels，就可以解決移動坐標軸和調(diào)整ticklabels與坐標軸的距離的問題。ps坐標軸上的小橫線博主嫌麻煩沒畫，只要位置找得好可以用ax.plot畫出來的。

???最后附上代碼。代碼含畫第一個不滿意的圖的方法，我給注釋掉了（在49-57行），想用傳統(tǒng)坐標軸的可以去掉注釋使用。

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 首先定義圖片尺寸，（900*600）
fig1 = plt.figure(figsize=(9, 6))
ax1 = Axes3D(fig1)
# 其次定義圖片輸出的視角
# ax1.view_init(elev=35, #仰角
#
# azim=60 #方位角
# )
ax1.view_init(elev=13, #仰角

azim=60 #方位角
)
# fake data
_x = np.arange(4)
_y = np.arange(4)
_xx, _yy = np.meshgrid(_x, _y)
x, y = _xx.ravel(), _yy.ravel()

DHRR = np.array([[0.5845, 0.0031, 0.051, 0.3871],
                 [0.0031, 0.0432, 0.0096, 0.0410],
                 [0.0510, 0.0096, 0.0590, 0.0338],
                 [0.3871, 0.0410, 0.0338, 0.3133]
                     ])
DHR = np.array([[0.5, 0.0, 0.0, 0.5],
                 [0.00, 0.0, 0.0, 0.0],
                 [0.0, 0.0, 0.0, 0.0],
                 [0.5, 0.0, 0.0, 0.5]
                     ])

dhr = DHR.ravel()
dhrr = DHRR.ravel()
bottom = np.zeros_like(dhrr)
width = 0.4
depth = 0.4
# 這里想給4個邊緣的角落不同的顏色，所以定義了一個矩陣，結合plt.cm來實現(xiàn)
colorA = plt.cm.pink(dhr.flatten())
ax1.bar3d(x - width/2, y - depth/2, bottom, width, depth, dhr, alpha=0.2, shade=False, color=colorA)
ax1.bar3d(x - width/2, y - depth/2, bottom, width, depth, dhrr, alpha=0.6, shade=True, color=colorA)
ax1.set_title('DH_Re')
ax1.set_zlabel('Probability ', size=10)
ax1.grid(False)
# 這一步很重要，直接關閉原來的軸，用ax1.set_axis_off()可以達到同樣的效果嘞
ax1.axis('off')
# 這里是老方法，去掉上面的ax1.axis('off')即可使用
# ax1.set_xticks(np.linspace(0, 3, 4))
# ax1.set_xticklabels([r'$\vert HH \rangle$', r'$\vert HV \rangle$', r'$\vert VH \rangle$',
#                      r'$\vert VH \rangle$'],
#                     size=16)
# ax1.set_yticks(np.linspace(0, 3, 4))
# ax1.set_yticklabels([r'$\vert HH \rangle$', r'$\vert HV \rangle$', r'$\vert VH \rangle$',
#                      r'$\vert VH \rangle$'],
#                     size=16)
# ax1.set_zbound(-0.5,0.5)
# y軸
ax1.plot([0.0, 1.25, 2.25, 3.25], [0, 0, 0, 0], [3.25, 3.25, 3.25, 3.25],
         zdir='y', color='#000000' )
# z軸
ax1.plot([0, 0, 0, 0, 0], [-0.5, -0.25, 0, 0.25, 0.5], [3.25, 3.25, 3.25, 3.25, 3.25],
         zdir='x', color='#000000')
# x軸
ax1.plot([0, 1.0, 2.0, 3.25], [0, 0, 0, 0], [3.25, 3.25, 3.25, 3.25],
         zdir='x', color='#000000')

# x軸上面的標記
ax1.text(3.6,0.2,0,r'$\vert HH \rangle$',size=10)
ax1.text(3.6,1.2,0,r'$\vert HV \rangle$',size=10)
ax1.text(3.6,2.2,0,r'$\vert VH \rangle$',size=10)
ax1.text(3.6,3.2,0,r'$\vert VV \rangle$',size=10)
# y軸上面的標記
ax1.text(3.2,3.6,0,r'$\vert HH \rangle$',size=10)
ax1.text(2.2,3.6,0,r'$\vert HV \rangle$',size=10)
ax1.text(1.2,3.6,0,r'$\vert VH \rangle$',size=10)
ax1.text(0.2,3.6,0,r'$\vert VV \rangle$',size=10)
# z軸上的標記
ax1.text(3.6,0,-0.5, '-0.5',size=10)
ax1.text(3.6,0,-0.4, '-0.4',size=10)
ax1.text(3.6,0,-0.3, '-0.3',size=10)
ax1.text(3.6,0,-0.2, '-0.2',size=10)
ax1.text(3.6,0,-0.1, '-0.1',size=10)
ax1.text(3.6,0,0, '0',size=10)
ax1.text(3.6,0,0.1, '0.1',size=10)
ax1.text(3.6,0,0.2,'0.2',size=10)
ax1.text(3.6,0,0.3, '0.3',size=10)
ax1.text(3.6,0,0.4,'0.4',size=10)
ax1.text(3.6,0,0.5, '0.5',size=10)
## 可以把具體的數(shù)值打在直方圖對應bar的頭上
a = np.linspace(0, 3, 4).astype('int8')
xs = np.linspace(0, 3, 4).astype('int8')
for n in xs:
    for m in a:
       tran_DHR = DHR[n, m]
       if tran_DHR != 0.00:
         ax1.text(n, m, tran_DHR + 0.03, tran_DHR, size=8, color='#FF0000' )
       tran_DHRR = DHRR[n, m]
       if abs(tran_DHRR) > 0.01:
         ax1.text(n, m, tran_DHRR + 0.03, tran_DHRR, size=8)

## 輸出圖片啦
plt.savefig('DHRE')
plt.show()

???最后附上上面代碼的效果圖！！！

文章來源地址http://www.zghlxwxcb.cn/news/detail-611785.html

到了這里，關于Python Axes3D自制坐標軸（沒辦法俺不知道咋移動坐標軸，但是不移動坐標軸畫圖太難看了）的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章，希望大家以后多多支持TOY模板網(wǎng)！