Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách xóa cột DataFrame bằng hàm 'pop' 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 trong DataFrame có thể xóa bằng nhiều phương pháp. Ta xem hàm pop 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 hàm 'pop':")
my_df.pop('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 hàm 'pop':
    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 pop xóa cột cụ thể.
  • Tên cột cần xóa truyền làm tham số cho pop.
  • 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ệ.

Đặc điểm của pop(): Trả về cột bị xóa (Series), sửa đổi DataFrame tại chỗ (in-place). Khác với del df['col'] không trả về giá trị, và df.drop('col', axis=1) trả về DataFrame mới (mặc định không in-place).