Menu
CoddyTech

Matriz de correlación

Parte de la sección Estadística para ML del Journey de Machine Learning de Coddy. Lección 31 de 64.

Elegir características significa comparar cada columna con el objetivo a la vez. df.corr(numeric_only=True) crea la matriz de correlación, omitiendo las columnas de texto:

laptops = pd.DataFrame({
    "model": ["Arc", "Bolt", "Cove", "Dune", "Echo", "Flux", "Gale"],
    "ram_gb": [8, 16, 8, 32, 16, 16, 32],
    "weight_kg": [2.3, 1.4, 2.0, 1.1, 1.8, 1.2, 1.6],
    "battery_h": [7, 9, 10, 13, 6, 11, 8],
    "price": [650, 1100, 700, 1600, 900, 1350, 1200],
})
corr = laptops.corr(numeric_only=True)

La matriz de correlación de cuatro columnas de laptops, representada como una cuadrícula de números. La diagonal contiene 1.00 en gris; se destaca la columna price: ram_gb 0.81, weight_kg -0.95, battery_h 0.70

Cada celda es la r de su fila y columna. Para encontrar la característica más fuerte, ordena la columna de precios por magnitud, ignorando los signos:

ranked = np.abs(corr["price"]).sort_values(ascending=False)
print(ranked.index[1])

Esto imprime:

weight_kg

El índice 0 es price en sí, en 1. Los portátiles más ligeros cuestan más: -0.95 es el vínculo más fuerte.

challenge icon

Desafío

Intermedio

Una compañía energética encuestó a 24 hogares en invierno. homes.csv tiene una columna de código home y las columnas numéricas area_m2, insulation_cm, residents, outside_temp_c y heating_kwh. El programa lee una columna objetivo.

  1. Imprime la columna objetivo de la matriz de correlación, redondeada a 2 decimales.
  2. Imprime Strongest:, la otra columna más fuertemente relacionada con ella y su r, como se muestra a continuación.

Para otra encuesta y el objetivo heating_kwh:

windows        0.72
floors         0.68
age_years      0.23
heating_kwh    1.00
Name: heating_kwh, dtype: float64
Strongest: windows (r = 0.72)

Pruébalo tú mismo

import numpy as np
import pandas as pd

homes = pd.read_csv("homes.csv")
target = input()

# Escribe tu código a continuación
quiz iconPonte a prueba

Esta lección incluye un breve cuestionario. Empieza la lección para responderlo y registrar tu progreso.

Todas las lecciones de Estadística para ML

Practica por tu cuenta: Compilador de Python online