Diagram Sebar dari Kolom

Termasuk bagian Data dan Plot dalam perjalanan Machine Learning di Coddy. Pelajaran 48 dari 60.

Apakah baterai yang lebih besar berarti jangkauan yang lebih jauh? Sebaran dua kolom menjawabnya sekilas, satu titik untuk setiap baris:

Diagram sebar berjudul Jangkauan Sepeda Listrik: tujuh sepeda listrik ditampilkan sebagai titik, kapasitas baterai dalam Wh pada sumbu mendatar dan jangkauan dalam km pada sumbu tegak, keduanya meningkat bersama dari 250 Wh dan 35 km hingga 750 Wh dan 110 km

Kolom mengisi posisi yang sebelumnya ditempati list, jadi ax.scatter menerima dua kolom:

bikes = pd.DataFrame({
    "battery_wh": [250, 300, 400, 500, 500, 625, 750],
    "range_km": [35, 45, 60, 65, 75, 95, 110],
})
fig, ax = plt.subplots()
ax.scatter(bikes["battery_wh"], bikes["range_km"])
ax.set_title("E-bike Range")
ax.set_xlabel("Battery (Wh)")
ax.set_ylabel("Range (km)")
plt.show()

Titik-titik yang naik ke kanan berarti kolom-kolom bertambah bersama: sebuah hubungan yang dapat dipelajari model. Sebaran titik yang tak beraturan berarti tidak ada hubungan.

ax.scatter tidak pernah melihat nama kolom, jadi kamu yang mengatur label sumbu.

Tantangan

Sedang

Seorang penjaga taman mengukur 12 pohon di trees.csv: age_years, trunk_cm, dan height_m. Mana yang cocok dengan tinggi, umur, atau lebar batang? Kode awal membaca salah satu dari dua nama itu ke dalam column dan menyiapkan diagram.

Gambar satu titik untuk setiap pohon: kolom tersebut pada sumbu mendatar, height_m pada sumbu tegak.
Beri label sumbu x dengan nama kolom tersebut.

Hasilmu seharusnya terlihat seperti diagram ini, untuk taman lain dengan crown_m:

Diagram sebar berjudul Ukuran Pohon, dengan crown_m pada sumbu mendatar dan Tinggi (m) pada sumbu tegak: sepuluh titik yang naik secara longgar, dari pohon setinggi sekitar 6 m dengan tajuk kecil hingga pohon setinggi 14 m dengan tajuk selebar 7.5 m

Coddy menjalankan kodemu untuk kedua kolom dan menjelaskan setiap diagram. Bandingkan keduanya: pita yang lebih rapat dan menanjak menjawab pertanyaan tersebut.

Coba sendiri

import pandas as pd
import matplotlib.pyplot as plt

trees = pd.read_csv("trees.csv")
column = input()

fig, ax = plt.subplots()
ax.set_title("Tree Size")
ax.set_ylabel("Height (m)")

# Tulis kodemu di bawah ini

# Jangan ubah baris di bawah ini
plt.show()