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

Xóa một hoặc nhiều cột trong DataFrame Python pandas (3 cách)

Giả sử bạn có DataFrame 3 cột và muốn xóa cột theo các cách khác nhau.

Dữ liệu mẫu

   one  two  three
0    1    2      3
1    4    5      6

Cách 1: del df['col']

import pandas as pd
data = [[1,2,3],[4,5,6]]
df = pd.DataFrame(data, columns=('one','two','three'))
print("Trước:", df)
del df['one']
print("Sau xóa 'one':\n", df)

Cách 2: df.pop('col') (trả về cột đã xóa)

df = pd.DataFrame(data, columns=('one','two','three'))
popped = df.pop('one')
print("Cột đã pop:", popped.tolist())
print("DataFrame còn lại:\n", df)

Cách 3: df.drop(columns=[...], inplace=True) (xóa nhiều cột)

df = pd.DataFrame(data, columns=('one','two','three'))
df.drop(columns=['one','two'], inplace=True)
print("Sau xóa 2 cột:\n", df)

Kết quả

Cách 1 & 2 - Xóa 1 cột 'one':
   two  three
0    2      3
1    5      6

Cách 3 - Xóa 2 cột 'one', 'two':
   three
0      3
1      6

So sánh:

  • del: Đơn giản, không trả về giá trị.
  • pop: Trả về Series cột đã xóa (hữu ích nếu cần dùng lại).
  • drop: Linh hoạt nhất, xóa nhiều cột/hàng, có inplace, errors='ignore'.