0

事前に計算された 2D ヒストグラムのようなデータセットをプロットしようとしています。線形/対数のz軸を持つ線形のy軸に対して対数のx軸をプロットする必要があります。ただし、 pcolor は最後の行と列を削除します。これは、最大のエネルギー値をプロットする必要があるため問題です。Imshow は、対数軸では機能しません。

適切にプロットするには、配列に NaN を埋め込む必要があると考えています。使用できる純粋なプロット ルーチンはありますか? ありがとう。

サンプルコード:

    # INITIALIZATION
    alpha_bounds = [0.0, 90.0]
    alpha_step = 15.0
    all_alphas = np.arange(alpha_bounds[0], alpha_bounds[1], alpha_step)
    beta_bounds = [0.0, 360.0]
    beta_step = 45.0
    all_betas = np.arange(beta_bounds[0], beta_bounds[1], beta_step)
    energy_bounds = [2e3, 5e6]
    n_energies = 15
    all_energies = np.logspace(
        np.log10(energy_bounds[0]), np.log10(energy_bounds[1]), n_energies)
    all_locations =\
        [(-52.5, 180.0, r - 1), (-77.5, 260.0, r - 1)]
    alts = np.linspace(
        70.0, 600.0, 500)

    # ARRAY TO GET ELOSS PER ALT, LOC, BETA, ALPHA, ENERGY
    # Changed np.zeros to np.ones for testing
    eloss_per_alt_per_process = np.ones(
         (len(all_locations),
          len(all_alphas), len(all_betas), len(all_energies),
          len(alts), len(processes)))

    # CODE HERE TO COUNT ELOSS PER ALT, LOC, BETA, ALPHA, ENERGY

    # SUM OVER TWO AXES
    eloss_per_alt_per_process[0, 0, 0, 0,
                              :, :] = np.sum(eloss_per_alt, axis=(1, 2))

    # PLOTTING
    ALT, E = np.meshgrid(np.array(all_energies), alts)
    eloss = np.transpose(
                eloss_per_alt_per_process[0, 0, 0, :, :, 2])
    if np.any(eloss):
                plt.figure()
                plt.pcolor(
                    ALT, E, eloss) #, norm=LogNorm()) #, vmin=1e-1,
                #    vmax=ncoll.max())
                plt.xscale('log')
    plt.show()
4

1 に答える 1

0

最初の質問: xとyの次元が z よりも 1 大きくなるように、メッシュのエッジを指定する必要があります。

matplotlib.colors.LogNorm2 番目の質問: のインスタンスをnormキーワードに与える必要があります。

簡単な例を次に示します。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import LogNorm

mean = [0, 0]
cov = [[1, 0.5],
       [0.5, 1]]

# draw some random numbers to show in the 2dhist
data = np.random.multivariate_normal(mean, cov, 100000)

# create the histogram
entries, xedges, yedges = np.histogram2d(data[:,0], data[:,1], [10, 10], [[-5, 5], [-5, 5]])

# create all combinations of x & y
x, y = np.meshgrid(xedges, yedges)

# plot, note that x and y are (m + 1) x (n + 1) arrays if entries is n x m
plt.pcolor(x, y, entries, norm=LogNorm(), cmap='afmhot')
plt.colorbar()
plt.tight_layout()
plt.savefig('2hist.png')

そして結果の画像： 結果

于 2015-03-13T10:34:19.043 に答える