Menu
CoddyTech

Korrelationsmatrix

Teil des Abschnitts Statistik für ML der Machine Learning-Journey von Coddy. Lektion 31 von 64.

Bei der Auswahl von Merkmalen werden alle Spalten gleichzeitig mit dem Ziel verglichen. df.corr(numeric_only=True) erstellt die Korrelationsmatrix und überspringt Textspalten:

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)

Die Korrelationsmatrix von vier Laptopspalten als Zahlenraster. Die Diagonale enthält 1.00 in Grau; die Spalte price ist hervorgehoben: ram_gb 0.81, weight_kg -0.95, battery_h 0.70

Jede Zelle ist das r ihrer Zeile und Spalte. Um das stärkste Merkmal zu finden, sortiere die Preisspalte nach Größe und ignoriere dabei die Vorzeichen:

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

Dies wird ausgegeben:

weight_kg

Index 0 ist price selbst, bei 1. Leichtere Laptops kosten mehr: -0.95 ist der stärkste Zusammenhang.

challenge icon

Aufgabe

Mittel

Ein Energieunternehmen befragte im Winter 24 Haushalte. homes.csv enthält einen home-Code und die Zahlenspalten area_m2, insulation_cm, residents, outside_temp_c und heating_kwh. Das Programm liest eine Zielspalte ein.

  1. Gib die Zielspalte der Korrelationsmatrix, auf 2 Dezimalstellen gerundet, aus.
  2. Gib Strongest:, die andere Spalte mit dem stärksten Zusammenhang dazu sowie ihr r wie unten aus.

Für eine andere Umfrage und das Ziel 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)

Probier es selbst

import numpy as np
import pandas as pd

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

# Schreibe deinen Code unten
quiz iconTeste dich selbst

Diese Lektion enthält ein kurzes Quiz. Starte die Lektion, um es zu beantworten und deinen Fortschritt zu speichern.

Alle Lektionen in Statistik für ML

Übe selbstständig: Online-Python-Compiler