Chọn cột

Hầu hết câu hỏi chỉ xoay quanh một cột: cân nặng trung bình, chú chó cao nhất. Đặt tên cột trong dấu ngoặc vuông để chọn cột đó:

dogs = pd.DataFrame({
    "name": ["Max", "Luna", "Bo", "Kira"],
    "kg": [30, 8, 22, 12],
    "cm": [61, 30, 55, 40],
})
print(dogs["kg"])
print(dogs["kg"].mean())

Lệnh này in ra một Series, được đặt tên theo cột của nó, rồi đến trọng lượng trung bình:

0    30
1     8
2    22
3    12
Name: kg, dtype: int64
18.0

Để chọn nhiều cột, hãy đặt danh sách tên cột trong dấu ngoặc vuông. Bạn sẽ nhận được một DataFrame nhỏ hơn, với các cột theo thứ tự bạn chọn:

print(dogs[["name", "cm"]])

Kết quả in ra:

   name  cm
0   Max  61
1  Luna  30
2    Bo  55
3  Kira  40

Việc chọn các cột giúp mô hình sau này lấy đúng các đặc trưng cần học.