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

Chọn nhiều cột DataFrame bằng regex và kiểu dữ liệu


Hướng dẫn chọn cột trong Pandas DataFrame: theo tên, regex, và kiểu dữ liệu.

1. Chọn cột đơn / nhiều cột

# Series
df["title"]
# DataFrame
df[["title"]]
# Nhiều cột
cols = ["title", "runtime", "vote_average", "vote_count"]
df[cols]

2. DataFrame.filter() - chọn theo tên cột

  • like: chứa chuỗi con. df.filter(like="title") → cột original_title, title.
  • regex: biểu thức chính quy. df.filter(regex="t$") → cột kết thúc bằng 't'.
  • items: danh sách tên cột chính xác.

3. select_dtypes() - chọn theo kiểu dữ liệu

# Chỉ int
df.select_dtypes(include="int")
# int và float
df.select_dtypes(include=["int64", "float"])
# Tất cả numeric
df.select_dtypes(include="number")
# Loại trừ object (string)
df.select_dtypes(exclude=["object"])

Lưu ý: Pandas không có kiểu "string", chuỗi là "object".

Kết luận

filter cho chọn theo tên, select_dtypes cho chọn theo kiểu dữ liệu.