이번 글에서는 polars DataFrame에서 특정 컬럼을 제외하는 방법을 살펴보겠습니다.
데이터
코드에 사용된 예제 데이터는 아래와 같이 만들었습니다.
1
2
3
4
5
6
7
8
import polars as pl
# 예제 데이터
df = pl.DataFrame({
"A": [1, 2, 3],
"B": [4, 5, 6],
"C": [7, 8, 9]
})
drop
drop을 사용해 제외 할 컬럼을 지정할 수 있습니다.
1
2
3
4
5
6
7
8
9
10
df.drop("B")
# ┌─────┬─────┐
# │ A ┆ C │
# │ --- ┆ --- │
# │ i64 ┆ i64 │
# ╞═════╪═════╡
# │ 1 ┆ 7 │
# │ 2 ┆ 8 │
# │ 3 ┆ 9 │
# └─────┴─────┘
pl.exclude
또한, select와 exclude를 사용해 제외 할 컬럼을 지정할 수 있습니다.
1
2
3
4
5
6
7
8
9
10
df.select(pl.exclude(["A", "C"]))
# ┌─────┐
# │ B │
# │ --- │
# │ i64 │
# ╞═════╡
# │ 4 │
# │ 5 │
# │ 6 │
# └─────┘