es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

  • es.davy.ai/
  • Programación /
  • Agregue marcas (por ejemplo, “x”) únicamente en ventanas específicas de los mapas de calor imshow.

Agregue marcas (por ejemplo, “x”) únicamente en ventanas específicas de los mapas de calor imshow.

19 Jun, 2023 Programación

¿Es posible anotar un mapa de calor imshow de tal manera que si el valor de un DataFrame de pandas es, por ejemplo, inferior a 3, se haga una marca “x” en esa ventana del mapa de calor específico?

Supongamos que tengo datos similares a este ejemplo:

d = {'col1': [1,2,1,5,3], 'col2': [3,4,1,5,2],'col3': [3,4,3,1,2],'col4': [3,4,1,2,5]}
df = pd.DataFrame(data=d)

fig = plt.figure(figsize=(6,4))
plot = plt.imshow(df)
ax = plot.axes
ax.invertyaxis()
ax.set_xticklabels(df.columns)

Vi que podemos anotar todas las ventanas del mapa de calor con los valores correspondientes, sin embargo, no puedo encontrar cómo anotar solo las ventanas debajo de mi límite dado y no con el valor correspondiente, sino con la marca “x”. ¿Puedo pedir sugerencias sobre si es posible con mapas de calor de matplotlib, por favor? Gracias.

Tags:  , , ,

Answer

  1. Avatar for davy.ai
    davy.ai

    Sí, es posible anotar ventanas específicas de un mapa de calor basándote en una condición utilizando matplotlib. Puedes usar bucles anidados para recorrer las filas y columnas del DataFrame y verificar si el valor es menor que 3. Si es así, puedes anotar la ventana correspondiente del mapa de calor con la marca “x” utilizando el método “text” del objeto del eje.

    Aquí hay un ejemplo de código que demuestra esto:

    import pandas as pd
import matplotlib.pyplot as plt

d = {'col1': [1,2,1,5,3], 'col2': [3,4,1,5,2],'col3': [3,4,3,1,2],'col4': [3,4,1,2,5]}
df = pd.DataFrame(data=d)

fig = plt.figure(figsize=(6,4))
plot = plt.imshow(df)
ax = plot.axes
ax.invert_yaxis()
ax.set_xticklabels(df.columns)

# anota las ventanas del mapa de calor con un valor menor a 3
for i in range(df.shape[0]):
    for j in range(df.shape[1]):
        if df.iloc[i,j] < 3:
            ax.text(j, i, "x", ha="center", va="center", color="black")

plt.show()

    Este código crea un mapa de calor del DataFrame y luego recorre cada celda del DataFrame y anota la ventana correspondiente del mapa de calor con la marca “x” si el valor de la celda es menor que 3. El resultado tendrá marcas “x” en las ventanas con valores menores a 3.

    Ten en cuenta que el método “text” toma las coordenadas x e y de la ventana del mapa de calor que se va a anotar, por lo que utilizamos los índices de fila (i) y columna (j) del DataFrame para calcular estas coordenadas. Los parámetros “ha” y “va” especifican la alineación horizontal y vertical de la anotación (“centro” en este caso) y el parámetro “color” especifica el color de la marca.

Comments are closed.