Cách xóa cột DataFrame trong Python

DataFrame là cấu trúc dữ liệu 2 chiều, lưu dạng bảng hàng và cột. Hình dung như bảng SQL hoặc Excel. Cột có thể xóa bằng nhiều phương pháp. Ta xem toán tử del nhận tên cột cần xóa làm tham số và xóa nó.

Ví dụ

import pandas as pd

my_data = {
    'ab': pd.Series([1, 8, 7], index=['a', 'b', 'c']),
    'cd': pd.Series([1, 2, 0, 9], index=['a', 'b', 'c', 'd']),
    'ef': pd.Series([56, 78, 32], index=['a', 'b', 'c']),
    'gh': pd.Series([66, 77, 88, 99], index=['a', 'b', 'c', 'd'])
}
my_df = pd.DataFrame(my_data)
print("DataFrame là:")
print(my_df)

print("Xóa cột dùng toán tử 'del':")
del my_df['cd']
print(my_df)

Đầu ra

DataFrame là:
    ab  cd   ef  gh
a  1.0   1  56.0  66
b  8.0   2  78.0  77
c  7.0   0  32.0  88
d  NaN   9  NaN  99

Xóa cột dùng toán tử 'del':
    ab   ef  gh
a  1.0  56.0  66
b  8.0  78.0  77
c  7.0  32.0  88
d  NaN  NaN  99

Giải thích

• Nhập thư viện cần thiết, đặt bí danh để dễ dùng.
• Tạo dictionary value gồm key-value, value là Series.
• Dictionary truyền làm tham số cho DataFrame trong pandas.
• Dùng từ khóa del xóa cột cụ thể.
• Tên cột cần xóa truyền làm tham số cho del.
• DataFrame mới in ra console.

Lưu ý - 'NaN' là 'Not a Number', nghĩa là [hàng, cột] cụ thể không có entry hợp lệ.

So sánh 3 cách xóa cột:

• del df['col'] - Xóa tại chỗ, không trả về giá trị, lỗi nếu cột không tồn tại
• df.pop('col') - Xóa tại chỗ, trả về Series cột đã xóa, lỗi nếu không tồn tại
• df.drop('col', axis=1) - Mặc định trả về DataFrame mới, inplace=True sửa tại chỗ, errors='ignore' không lỗi nếu thiếu cột