15

電場と磁場における原子の相互作用をモデル化するコードを書いています。この一部では、特定の高さでの相互作用ポテンシャルの 3D プロットを作成する必要があります。これらのプロットを生成する完全なコードは非常に長く、いくつかのモジュールに分かれていますが、関連するプロット部分は次のとおりです。

# Function to plot 'PlotValues' at a height 'z'
def Plot_AtHeight(self, PlotValues, z=500, ReturnFig=False, ShowTime=False):

    # Calls out to the relevant function to calculate the values and return
    # these as an array
    PlotArray = self.Get_AtHeight(PlotValues, z)

    pylab.rcParams.update( \
        {'axes.labelsize': 18,
         'text.fontsize': 18,
         'xtick.labelsize': 18,
         'ytick.labelsize': 18
         })

    fig = pylab.figure()
    ax = Axes3D(fig)

    # Make the arrays of the points at which the values are calculated
    X, Y = np.mgrid[Xmin:Xmax:complex(0,Xpoints),
                   Ymin:Ymax:complex(0,Ypoints)]

    ax.plot_surface(X, Y, PlotArray, cmap=cm.jet)

    ax.set_xlabel('Position, x (nm)')
    ax.set_ylabel('Position, y (nm)')

    if PlotValues   == 'B':           ax.set_zlabel('Field Strength (G)',         fontsize=18)
    elif PlotValues == 'E':           ax.set_zlabel('Field Strength (V/m)',       fontsize=18)
    elif PlotValues == 'U_Stark':     ax.set_zlabel('Stark Interaction (J)',      fontsize=18)
    elif PlotValues == 'U_Zeeman':    ax.set_zlabel('Zeeman Interaction (J)',     fontsize=18)
    elif PlotValues == 'U':           ax.set_zlabel('Interaction Potential (J)',  fontsize=18)
    elif PlotValues == 'U_Stark_mK':  ax.set_zlabel('Stark Interaction (mK)',     fontsize=18)
    elif PlotValues == 'U_Zeeman_mK': ax.set_zlabel('Zeeman Interaction (mK)',    fontsize=18)
    elif PlotValues == 'U_mK':        ax.set_zlabel('Interaction Potential (mK)', fontsize=18)

    # If we are not in a time averaged environment then display the current
    # time (in ns) as the title to 1 decimal place.
    if not self.TimeAveraged and ShowTime:
        TimeStr = str(time*10**9)
        try:
            TimeTo1dp = '.'.join([TimeStr.split('.')[0], TimeStr.split('.')[1][0]])
        except:
            TimeTo1dp = TimeStr
        ax.set_title("t = %sns" % TimeTo1dp, fontsize=18)

    if not ReturnFig: pylab.show()
    elif ReturnFig: return fig

これが返すプロットの例は次のとおりです。

作成されたプロットの 1 つ

軸ラベルと目盛りが少し乱雑になっていることがわかります。特に、誰かが画像の下部が切り取られるのを止める方法を知っていることを望んでいました (つまり、1000 のすべてがクリアになるように)。私は多くの角度でこの問題を抱えています。軸ラベルが切り取られることもあれば、目盛りが表示されることもありますが、基本的にウィンドウ python が開いてプロットを表示および保存するのは十分に大きくないようであり、それを拡大すると画像全体がスケーリングされるため、ラベル/ダニはまだカットされています。

フォントサイズを小さくしたり、ラベルを削除したりすることについては言及しないでください。これはレポートに含まれるため、これらは修正されます。

ありがとう。

4

1 に答える 1

20

たとえば、次のようにして、表示の「距離」をプロットに設定できます。

ax.dist = 13

于 2014-03-05T16:36:00.917 に答える