如何在matplotlib中添加第二个x轴

Question

我有一个非常简单的问题。我需要在我的绘图上有第二个x轴，我希望这个轴有一定数量的抽搐，对应于第一个轴的某个位置。

让我们试试一个例子。在这里，我正在绘制暗物质质量作为扩展因子的函数，定义为1 /（1 + z），范围从0到1.

semilogy(1/(1+z),mass_acc_massive,'-',label='DM')
xlim(0,1)
ylim(1e8,5e12)

我想在我的图的顶部有另一个x轴，显示扩展因子的某些值的相应z。那可能吗？如果是的话，我怎么能有xtics ax

Answer 1

我从@Dhara的答案中的评论中得到一个提示，听起来你想要通过旧x轴到新x轴的函数设置new_tick_locations列表。下面的tick_function接收一个数量级的点，将它们映射到一个新值并格式化它们：

import numpy as np
import matplotlib.pyplot as plt

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

X = np.linspace(0,1,1000)
Y = np.cos(X*20)

ax1.plot(X,Y)
ax1.set_xlabel(r"Original x-axis: $X$")

new_tick_locations = np.array([.2, .5, .9])

def tick_function(X):
    V = 1/(1+X)
    return ["%.3f" % z for z in V]

ax2.set_xlim(ax1.get_xlim())
ax2.set_xticks(new_tick_locations)
ax2.set_xticklabels(tick_function(new_tick_locations))
ax2.set_xlabel(r"Modified x-axis: $1/(1+X)$")
plt.show()

Answer 2

您可以使用twiny来创建2个x轴刻度。例如：

import numpy as np
import matplotlib.pyplot as plt

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

a = np.cos(2*np.pi*np.linspace(0, 1, 60.))

ax1.plot(range(60), a)
ax2.plot(range(100), np.ones(100)) # Create a dummy plot
ax2.cla()
plt.show()

参考：http://matplotlib.sourceforge.net/faq/howto_faq.html#multiple-y-axis-scales

输出：

Answer 3

如果您希望您的上轴是下轴刻度值的函数：

import matplotlib.pyplot as plt

fig, ax1 = plt.subplots()

ax1 = fig.add_subplot(111)

ax1.plot(range(5), range(5))

ax1.grid(True)

ax2 = ax1.twiny()
ax1Xs = ax1.get_xticks()

ax2Xs = []
for X in ax1Xs:
    ax2Xs.append(X * 2)

ax2.set_xticks(ax1Xs)
ax2.set_xbound(ax1.get_xbound())
ax2.set_xticklabels(ax2Xs)

title = ax1.set_title("Upper x-axis ticks are lower x-axis ticks doubled!")
title.set_y(1.1)
fig.subplots_adjust(top=0.85)

fig.savefig("1.png")

给出：

Answer 4

在Dhara的回答评论中回答你的问题：“我希望在第二个x轴上这些抽搐：（7,8,99）对应于x轴位置10,30,40。这是可能的吗？在某种程度上？“ 是的，是的。

import numpy as np
import matplotlib.pyplot as plt

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

a = np.cos(2*np.pi*np.linspace(0, 1, 60.))
ax1.plot(range(60), a)

ax1.set_xlim(0, 60)
ax1.set_xlabel("x")
ax1.set_ylabel("y")

ax2 = ax1.twiny()
ax2.set_xlabel("x-transformed")
ax2.set_xlim(0, 60)
ax2.set_xticks([10, 30, 40])
ax2.set_xticklabels(['7','8','99'])

plt.show()

你会得到：

Answer 5

由于声誉低，我不得不将此作为答案而不是评论发布。 我和Matteo有类似的问题。不同之处在于我没有从第一个x轴到第二个x轴的地图，只有x值本身。所以我想直接在我的第二个x轴上设置数据，而不是刻度线，但是，没有axes.set_xdata。我能够使用Dhara的答案进行修改：

ax2.lines = []

而不是使用：

ax2.cla()

使用时也从ax1清除了我的情节。

Answer 6

从matplotlib 3.1开始，您可以使用ax.secondary_xaxis

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(1,13, num=301)
y = (np.sin(x)+1.01)*3000

# Define function and its inverse
f = lambda x: 1/(1+x)
g = lambda x: 1/x-1

fig, ax = plt.subplots()
ax.semilogy(x, y, label='DM')

ax2 = ax.secondary_xaxis("top", functions=(f,g))

ax2.set_xlabel("1/(x+1)")
ax.set_xlabel("x")
plt.show()

enter image description here