Menu
CoddyTech

Matrice di correlazione

Fa parte della sezione Statistica per il machine learning del percorso Machine Learning di Coddy. Lezione 31 di 64.

Scegliere le caratteristiche significa confrontare contemporaneamente ogni colonna con il valore target. df.corr(numeric_only=True) crea la matrice di correlazione, ignorando le colonne di testo:

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 matrice di correlazione di quattro colonne di laptop, rappresentata come una griglia di numeri. La diagonale contiene 1.00 in grigio; la colonna price è evidenziata: ram_gb 0.81, weight_kg -0.95, battery_h 0.70

Ogni cella è il valore di r della sua riga e colonna. Per trovare la caratteristica più forte, ordina la colonna price in base alla grandezza, ignorando i segni:

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

Questo stampa:

weight_kg

L'indice 0 è price stesso, a 1. I laptop più leggeri costano di più: -0.95 è il legame più forte.

challenge icon

Sfida

Medio

Un’azienda energetica ha intervistato 24 abitazioni in inverno. homes.csv contiene un codice home e le colonne numeriche area_m2, insulation_cm, residents, outside_temp_c e heating_kwh. Il programma legge una colonna obiettivo.

  1. Stampa la colonna obiettivo della matrice di correlazione, arrotondata a 2 decimali.
  2. Stampa Strongest:, l’altra colonna più fortemente correlata a essa e il suo r, come mostrato sotto.

Per un’altra indagine e la colonna obiettivo 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)

Provalo tu

import numpy as np
import pandas as pd

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

# Scrivi il tuo codice qui sotto
quiz iconMettiti alla prova

Questa lezione include un breve quiz. Inizia la lezione per rispondere e tenere traccia dei tuoi progressi.

Tutte le lezioni di Statistica per il machine learning

Esercitati da solo: Compilatore Python online