如何用matplotlib正確創建輪廓？

Question

我具有下面給出的形式的數據（link）：

Y X 0 X 10 X 20 
15 4.83 4.91 4.99 
20 4.58 4.65 4.73 
25 4.43 4.49 4.56 

我試圖繪製X 標籤值的輪廓，在這些情況下爲0，10，20（如在代碼z），其中X軸爲X值，Y軸爲Y值。使用代碼創建地塊：

import numpy as np 
import re 
from matplotlib.backends.backend_pdf import PdfPages 
import matplotlib.pyplot as plt 
import matplotlib.cm as cm 

with open('contour.dat', "r") as data: 
    while True: 
     line = data.readline() 
     if not line.startswith('#'): 
      break 
    data_header = [i for i in line.strip().split('\t') if i] 
_data_ = np.genfromtxt('contour.dat', skiprows = 2, delimiter = '\t') 
x = _data_[:, 0] 
y = _data_[:, 1:] 


y_n = [] 
for i in range(len(data_header)): 
    if data_header[i][0] == 'X': 
     y_n = np.int_(np.append(y_n, i)) 
y_index = [data_header[i] for i in y_n] 
z = [] 
for i in range(0, len(data_header)): 
    z = np.append(z, re.findall(r"[-+]?\d*\.\d+|\d+", data_header[i])) 
z = z.reshape(len(z), 1) 
z = np.tile(z, (len(x), len(x))) 


with PdfPages('./on_tau.pdf') as p_tau: 
    _p_vs_tau_ = plt.figure(figsize=(5, 5)) 
    _p_vs_tau_.clf() 
    p_vs_tau = plt.subplot(111) 
    x, y = np.meshgrid(x, y) 
    surf = plt.contourf(y, x, z, 30, rstride=1, cstride=1, cmap=cm.gist_heat, 
          linewidth=0, antialiased=False, alpha = 1.0) 
    surf1 = plt.contour(y, x, z, 30, colors = '#000000', 
          linewidths=0.5, antialiased=False, alpha = 1.0) 
    plt.clabel(surf1, inline=1, fontsize=6) 
    p_tau.savefig(bbox_inches='tight') 
    plt.close() 

但是我沒有得到正確的輪廓。我相信我在某個地方做錯了什麼，可能是重複z值。如何正確獲取輪廓？

Answer 1

這是否產生你所尋找的輪廓？從Z = np.tile語句修改代碼（現在註釋掉）：

#z = np.tile(z, (len(x), len(x))) 
xm = np.tile(x,11) 
xm = np.reshape(xm,(11,10)).T 
zm = np.tile(z,10).T 

with PdfPages('./on_tau.pdf') as p_tau: 
    _p_vs_tau_ = plt.figure(figsize=(5, 5)) 
    _p_vs_tau_.clf() 
    p_vs_tau = plt.subplot(111) 
    #x, y = np.meshgrid(x, y) 
    surf = plt.contourf(y, xm, zm, 30, rstride=1, cstride=1, cmap=cm.gist_heat, 
         linewidth=0, antialiased=False, alpha = 1.0) 
    surf1 = plt.contour(y, xm, zm, 30, colors = '#000000', 
         linewidths=0.5, antialiased=False, alpha = 1.0) 
    plt.clabel(surf1, inline=1, fontsize=6) 
    p_tau.savefig(bbox_inches='tight') 
    plt.close()