python - Python 二维直方图

Question

我正在尝试使用这些代码在 Python 中绘制二维直方图

from math import *
import pylab as p
import matplotlib.pyplot as plt
import numpy as np

x=part.points[:,0]
y=part.points[:,1]
z=part.points[:,2]

H, xedges, yedges = np.histogram2d(x, y, bins=(128,128))
H.shape, xedges.shape, yedges.shape

extent = [yedges[0], yedges[-1], xedges[-1], xedges[0]]

plt.imshow(H, extent=extent, interpolation='nearest')

plt.colorbar()
plt.xlabel("x")
plt.ylabel("y")
plt.show()

一切正常:我有一个颜色条代表每个单元格中的计数。问题是我想要计数日志，但函数 histrogram2d 没有任何选项。

Answer 1

我想你可以简单地做

H_log = np.log(H)
…
plt.imshow(H_log,…)

(假设您没有空计数)。

如果您想要 3D 条形图，您可以调整 example在 Matplotlib 文档中提供。

更一般地说，我衷心建议您查看非常有用的 Matplotlib gallery ，当您正在寻找一些特定的图形功能时。