如何用最后一行/列和對數軸繪制matplotlib pcolor？

Question

我一直在嘗試繪制預先計算的類似於2D直方圖的數據集。 我必須將對數x軸相對於線性y軸與線性/對數z軸作圖。 但是，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()

Answer 1

第一個問題：您需要給網格的邊，使x和y的尺寸比z大一。

第二個問題：您需要給norm關鍵字提供一個matplotlib.colors.LogNorm實例。

這里是一個簡化的例子：

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')

以及得到的圖像：