Como posso obter um valor de specific (x, y) no mapa de contorno python

Dec 08 2020

Um exemplo em https://www.tutorialspoint.com/matplotlib/matplotlib_contour_plot.htm

import numpy as np
import matplotlib.pyplot as plt
xlist = np.linspace(-3.0, 3.0, 100)
ylist = np.linspace(-3.0, 3.0, 100)
X, Y = np.meshgrid(xlist, ylist)
Z = np.sqrt(X**2 + Y**2)
fig,ax=plt.subplots(1,1)
cp = ax.contourf(X, Y, Z)
fig.colorbar(cp) # Add a colorbar to a plot
ax.set_title('Filled Contours Plot')
#ax.set_xlabel('x (cm)')
ax.set_ylabel('y (cm)')
plt.show()

Agora, temos um mapa de contorno denominado cp, como posso obter o valor z do ponto (x, y) no mapa de contorno cp ???

Obrigado.

Respostas

2 PatrickArtner Dec 08 2020 at 14:45

A visualização é apenas uma representação sofisticada de seus dados em X, Ye Z.

Você já tem os valores calculados, você pode simplesmente procurá-los:

import numpy as np
import matplotlib.pyplot as plt

xlist = np.linspace(-3.0, 3.0, 100)
ylist = np.linspace(-3.0, 3.0, 100)
X, Y = np.meshgrid(xlist, ylist)
Z = np.sqrt(X**2 + Y**2)

x = 42  # xlist has 100 entries
y = 21  # ylist has 100 entries

print(f"x at pos {x} is {xlist[x]}", f"y at pos {y} is {ylist[y]}",
      f"z value at that place is {Z[x][y]}", sep="\n")

Resultado:

x at pos 42 is -0.4545454545454546
y at pos 21 is -1.7272727272727273
z value at that place is 1.7860802458535001

Se você deseja pesquisar um determinado x,y-valor, encontre o índice de closes em xliste yliste obtenha seu valor de Z.

xv = 1.2
yv = -0.7

x_pos = min(p for p in range(100) if xlist[p] >= xv) # could do something better
y_pos = min(p for p in range(100) if ylist[p] >= yv) # using np.where or such

# or use any other metric to get the "closest" point, f.e.
# d_x, x_pos = min( (abs(v), p) for p,v in enumerate(xlist - np.array([xv]*100)))
# d_y, y_pos = min( (abs(v), p) for p,v in enumerate(ylist - np.array([yv]*100)))

print(f"x at pos {x_pos} is {xlist[x_pos]} (looking for {xv})",
      f"y at pos {y_pos} is {ylist[y_pos]} (looking for {yv})",
      f"z value at that place is {Z[x_pos][y_pos]}", sep="\n") 

Resultado:

x at pos 70 is 1.2424242424242422 (looking for 1.2)
y at pos 38 is -0.6969696969696968 (looking for -0.7)
z value at that place is 1.4245647604294736

# x at pos 69 is 1.1818181818181817 (looking for 1.2)
# y at pos 38 is -0.6969696969696968 (looking for -0.7)
# z value at that place is 1.3720280512329417

Se você ainda quer ler diretamente do gráfico, tente: Como extrair pontos de um gráfico?