5

我有一个对数分布的大型数据集。我想制作一张热图,所以我做了一个 2D 直方图并将其传递给 implot。因为数据是对数的,所以我将数据的对数传递给直方图。但是,当我制作绘图时,我希望恢复轴(即 10^hist bin 值)和日志轴。如果我将轴设置为 log 样式,那么图像看起来都是倾斜的。当我将数据传递给直方图时,数据已经“记录”了,所以我不希望图像受到影响,只是轴。所以,在下面的例子中,我想要左边的图像,右边的轴。

我想我可以用一个假的重叠轴来做,但如果有更好的方法,我不喜欢做那种事情......

在此处输入图像描述

import numpy as np
import matplotlib.pyplot as plt

x=10**np.random.random(10000)*5
y=10**np.random.random(10000)*5

samps, xedges, yedges = np.histogram2d(np.log10(y), np.log10(x),     bins=50)    

ax = plt.subplot(121)

plt.imshow(samps, extent=[0,5,0,5])
plt.xlabel('Log10 X')
plt.ylabel('Log10 Y')

ax = plt.subplot(122)    
plt.imshow(samps, extent=[10**0,10**5,10**0,10**5])
plt.xlabel('X')
plt.ylabel('Y')
plt.xscale('log')
plt.yscale('log')
plt.show()
4

2 回答 2

7

您需要使用自定义格式化程序。这是 matplotlib 文档中的一个示例: https ://matplotlib.org/examples/pylab_examples/custom_ticker1.html

我倾向于FuncFormatter像示例一样使用。主要技巧是您的函数需要接受参数xpos. 老实说,我不知道这pos是为了什么。甚至可能不是故意的,但是您可以将FuncFormatter其用作装饰器,这就是我在下面所做的:

%matplotlib inline
import numpy as np
import matplotlib.pyplot as plt

@plt.FuncFormatter
def fake_log(x, pos):
    'The two args are the value and tick position'
    return r'$10^{%d}$' % (x)

x=10**np.random.random(10000)*5
y=10**np.random.random(10000)*5

samps, xedges, yedges = np.histogram2d(np.log10(y), np.log10(x), bins=50)    

fig, (ax1) = plt.subplots()
ax1.imshow(samps, extent=[0, 5, 0, 5])
ax1.xaxis.set_major_formatter(fake_log)
ax1.yaxis.set_major_formatter(fake_log)
ax1.set_xlabel('X')
ax1.set_ylabel('Y')

在此处输入图像描述

于 2018-02-19T04:29:28.687 回答
1

如果您只想更改标签,您可以通过plt.gca().set_xticklabels和直接访问这些标签plt.gca().set_yticklabels。这是一个更改这些_text属性的简单示例。

import numpy as np
import matplotlib.pyplot as plt

x = 10 ** np.random.random(10000) * 5
y = 10 ** np.random.random(10000) * 5

samps, xedges, yedges = np.histogram2d(np.log10(y), np.log10(x), bins=50)

plt.subplot(121)

p = plt.imshow(samps, extent=[0, 5, 0, 5])
plt.xlabel('Log10 X')
plt.ylabel('Log10 Y')

plt.subplot(122)

p = plt.imshow(samps, extent=[0, 5, 0, 5])

# The label handling stuff starts here
plt.pause(0.5)  # Needed to make sure the drawing finished being created
plt.xlabel('X')
plt.ylabel('Y')
plt.draw()

ax = plt.gca()
lbx = ax.get_xticklabels()
lby = ax.get_yticklabels()
for i in range(len(lbx)):
    lbx[i]._text = r'$10^' + lbx[i]._text + '$'
for i in range(len(lby)):
    lby[i]._text = r'$10^' + lby[i]._text + '$'
ax.set_xticklabels(lbx)
ax.set_yticklabels(lby)

plt.show()
于 2018-02-19T11:06:59.570 回答