Cách thêm trục x thứ hai vào matplotlib

Question

Tôi có một câu hỏi rất đơn giản. Tôi cần phải có một trục x thứ hai trên cốt truyện của tôi và tôi muốn rằng trục này có một số lượng nhất định của tics tương ứng với vị trí nhất định của trục đầu tiên.

Hãy thử một ví dụ. Ở đây tôi đang âm mưu thực hiện khối lượng vật chất tối là một chức năng của hệ số mở rộng, định nghĩa là 1/(1 + z), mà dao động từ 0 đến 1.

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

Tôi muốn có một trục x, trên đầu cốt truyện của tôi, hiển thị z tương ứng cho một số giá trị của hệ số mở rộng. Điều đó có thể không? Nếu có, làm thế nào tôi có thể có xtics axe

Answer 1

Tôi đang tham gia một gợi ý từ những ý kiến ​​trong @ Dhara của câu trả lời, có vẻ như bạn muốn thiết lập một danh sách các new_tick_locations bởi một hàm từ trục x cũ đến trục x mới. Các tick_function dưới đây mất trong một mảng NumPy điểm, bản đồ chúng vào một giá trị mới và định dạng chúng:

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

Bạn có thể sử dụng song sinh để tạo 2 thang trục x. Ví dụ:

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

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

Output:

Answer 3

Trả lời câu hỏi của bạn trong ý kiến ​​trả lời Dhara của: "tôi sẽ thích trên x thứ hai -axis các tics: (7,8,99) tương ứng với vị trí trục x 10, 30, 40. Có thể theo một cách nào đó không? " Vâng, đúng vậy.

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

Bạn sẽ nhận được:

Answer 4

Nếu Bạn muốn trục trên của bạn là một chức năng của trục dưới tick-giá trị:

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

Cung cấp:

Answer 5

Tôi buộc phải đăng bài này dưới dạng n trả lời thay vì nhận xét do danh tiếng thấp. Tôi gặp vấn đề tương tự với Matteo. Sự khác biệt là tôi không có bản đồ từ trục x đầu tiên của tôi đến trục x thứ hai của tôi, chỉ có các giá trị x. Vì vậy, tôi muốn đặt dữ liệu trên trục x thứ hai của mình trực tiếp, chứ không phải là ve, tuy nhiên, không có axes.set_xdata. Tôi đã có thể sử dụng câu trả lời Dhara để làm điều này với một sự sửa đổi:

ax2.lines = [] 

thay vì sử dụng:

ax2.cla() 

Khi sử dụng cũng xóa cốt truyện của tôi từ ax1.