1

我有一行数据,我想在不使用 twiny() 的情况下对其进行热映射,因为它会有对齐问题。我在这个网站上搜索了很多,这就是我到目前为止所达到的:

带有单行热图的 imshow

import numpy as np
import matplotlib.pyplot as plt

x = "0.2, 0.3, 0.4, 0.5, 0.6".split(",")
y = "180, 175, 170, 169, 150".split(",")
z = [[5000, 4800, 4500, 4450, 4300]]

fig, ax1 = plt.subplots()

image = z

im = ax1.imshow(image, cmap=plt.cm.Blues, interpolation='nearest')
plt.colorbar(im)


ax1.set_xticks(np.arange(len(x)), minor=False)

ax1.set_xticklabels(x, minor=False)
#ax1.set_yticklabels(y, minor=False)

ax1.tick_params(labelbottom='on',labeltop='on', labelleft="off")



plt.show()

如您所见,上轴与下轴具有完全相同的文本。我想要的是放在y上面的轴上。

提前致谢 :)

4

4 回答 4

4

看起来twiny纵横比设置为equal的轴不想生活在同一个轴上。对我来说,这似乎是一个错误,但可能有一个解释。

所以,让我们通过在彼此之上绘制两个轴来解决这个问题。这并不像听起来那么简单,因为如果两个子图位于同一位置,matplotlib则将它们解释为同一个。但是,使用add_plot不存在这样的问题。

import numpy as np
import matplotlib.pyplot as plt

x = "0.2, 0.3, 0.4, 0.5, 0.6".split(",")
y = "180, 175, 170, 169, 150".split(",")
z = [[5000, 4800, 4500, 4450, 4300]]

fig = plt.figure()
ax1 = fig.add_subplot(111)

image = z

im = ax1.imshow(image, cmap=plt.cm.Blues, interpolation='nearest')
plt.colorbar(im)

ax1.set_xticks(np.arange(len(x)), minor=False)
ax1.set_xticklabels(x, minor=False)
ax1.tick_params(labelbottom='on',labeltop='off', labelleft="off", 
    top='off', left='off', right='off')

# create another axes on the same position:
# - create second axes on top of the first one without background
# - make the background invisible
# - set the x scale according to that of `ax1`
# - set the top ticks on and everything else off
# - set the size according to the size of `ax1`
ax2 = fig.add_axes(ax1.get_position(), frameon=False)
ax2.tick_params(labelbottom='off',labeltop='on', labelleft="off", labelright='off',
    bottom='off', left='off', right='off')
ax2.set_xlim(ax1.get_xlim())
ax2.set_xticks(np.arange(len(y)))
ax2.set_xticklabels(y, minor=False)

plt.draw()
ax2.set_position(ax1.get_position())

plt.draw()
plt.show()

在theplt.draw()之前需要set_position,否则会由于使用方面而get_position返回错误的位置。(这可能就是失败的原因。)ax1equaltwiny

在此处输入图像描述


如果您需要多行,则解决方案没有什么不同:

import numpy as np
import matplotlib.pyplot as plt

x = "0.2, 0.3, 0.4, 0.5, 0.6".split(",")
y = "180, 175, 170, 169, 150".split(",")
z = [[5000, 4800, 4500, 4450, 4300]]

numRows = 8

fig, subaxes = plt.subplots(nrows=numRows, ncols=1)
axeslist = subaxes.flatten()

for ax in axeslist:
    im = ax.imshow(z, cmap=plt.cm.Blues, interpolation='nearest')

    ax.tick_params(labelbottom='off',labeltop='off', labelleft="off", labelright='off',
                   bottom='off', top='off', left='off', right='off')
    if ax == axeslist[0]:
        ax.set_title('Avg. (s)\n', size=13)
    elif ax == axeslist[-1]:
        ax.tick_params(bottom='on', labelbottom='on')
        ax.set_xticks(range(len(x)))
        ax.set_xticklabels(x)

# reserve some space between the subplots
fig.subplots_adjust(hspace=0.07*(numRows-1))

# create the overlay images, add them as extra properties of the original images
for ax in axeslist:
    axnew = fig.add_axes(ax.get_position(), frameon=False)
    axnew.tick_params(labelbottom='off',labeltop='on', labelleft="off", labelright='off',
                   bottom='off', top='on', left='off', right='off')
    axnew.set_xlim(ax.get_xlim())
    axnew.set_xticks(range(len(y)))
    axnew.set_xticklabels(y)
    ax.extra_axes = axnew

# update the secondary axes positions
# draw() only if there was something changed (important!)
def update_secondary(event=None):
    position_changed = False
    for ax in axeslist:
        if ax.extra_axes.get_position().bounds == ax.get_position().bounds:
            continue
        position_changed = True
        ax.extra_axes.set_position(ax.get_position())

    if position_changed:
        plt.draw()

# register the secondary axes updater as a callback
fig.canvas.mpl_connect('draw_event', update_secondary)

# make sure everything is drawn
plt.draw()    

在此处输入图像描述

由于必须在绘制完其他所有内容后执行覆盖更新,因此这里由draw_event后端完成。结果是,由于某种原因重绘图像后,叠加层被重新调整,如果任何位置发生变化,场景就会被重绘。

这有效,但并不漂亮。

于 2014-07-13T15:38:07.117 回答
1

好吧,我通过使用文本而不是更改轴值找到了替代解决方案。

imshow 没有 twiny 的单个顶部和底部标签

import numpy as np
import matplotlib.pyplot as plt

font = {'family' : 'sans-serif',
        'color'  : 'k',
        'weight' : 'normal',
        'size'   : 12,
        }

x = "0.2, 0.3, 0.4, 0.5, 0.6".split(",")
y = "180, 175, 170, 169, 150".split(",")
z = [[5000, 4800, 4500, 4450, 4300]]

fig, ax1 = plt.subplots()
#fig, ax1 = plt.subplots()

image = z

im = ax1.imshow(z, cmap=plt.cm.Blues, interpolation='nearest')
xticks = ax1.get_xticks()

top_lables_width_spacings = 0.83
top_lables_hight_spacings = -.53

for i in range(len(y)):
    ax1.text(xticks[i] + top_lables_width_spacings, top_lables_hight_spacings, y[i], fontdict=font)
#ax1.set_aspect('auto')

fig.colorbar(im, orientation='vertical')

ax1.set_xticks(np.arange(len(x)), minor=False)
ax1.set_xticklabels(x, minor=False)

ax1.tick_params(labelbottom='on',labeltop='off', labelleft="off")

ax1.set_title('$\eta$\n', size=17)      # represents the top axes label
plt.xlabel(r'$\theta$', size=17)                        # represents the bottom axes label
plt.show()
于 2014-07-13T13:52:51.513 回答
1

使用twin()from mpl_toolkits.axes_grid1 import host_subplot

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA
import numpy as np

x = "0.2, 0.3, 0.4, 0.5, 0.6".split(",")
y = "180, 175, 170, 169, 150".split(",")
z = [[5000, 4800, 4500, 4450, 4300]]

#fig, ax1 = plt.subplots()
ax1 = host_subplot(111, axes_class=AA.Axes)

ax2 = ax1.twin()
image = z

im = ax1.imshow(image, cmap=plt.cm.Blues, interpolation='nearest')
plt.colorbar(im)


ax1.set_xticks(np.arange(len(x)), minor=False)
ax2.set_xticks(np.arange(len(y)), minor=False)

#ax1.tick_params(labelbottom='on',labeltop='on', labelleft="off")
#ax2.tick_params(labelbottom='on',labeltop='on', labelleft="off")

ax1.set_yticklabels([])
ax2.set_yticklabels([])

ax1.tick_params(labelbottom='on',labeltop='on', labelleft="off")


plt.show()
于 2014-07-13T15:40:42.923 回答
0

如果我做对了,我想你只需要将 x 更改为 y:

ax1.set_xticklabels(y, minor=False)

如果我这样做,我会在图的顶部和底部看到列表 y 中的标签。

于 2014-07-13T13:45:38.933 回答