У меня есть следующий фрейм данных поляр, и я хотел динамически выбрать последний столбец.
>>> import polars as pl
>>>
>>> df = pl.DataFrame({
... "col1": [1, 2],
... "col2": ["2", "3"],
... "col3": [3, 4]
... })
>>>
>>> df
shape: (2, 3)
┌──────┬──────┬──────┐
│ col1 ┆ col2 ┆ col3 │
│ --- ┆ --- ┆ --- │
│ i64 ┆ str ┆ i64 │
╞══════╪══════╪══════╡
│ 1 ┆ 2 ┆ 3 │
│ 2 ┆ 3 ┆ 4 │
└──────┴──────┴──────┘
>>> # How to select col3? which is the last column in df
Как я могу сделать это в полярах?. Я могу сделать df.iloc[:,-1:], чтобы выбрать последний столбец, если это фрейм данных pandas.
Дополнительная информация:
>>> import sys
>>> sys.version_info
sys.version_info(major=3, minor=11, micro=0, releaselevel='final', serial=0)
>>> import polars
>>> polars.__version__
'0.18.3'






Чтобы помочь в подобных операциях, недавно был представлен модуль polars.selectors . Вы можете просто использовать last из этого модуля:
df.select(cs.last())
shape: (2, 1)
┌──────┐
│ col3 │
│ --- │
│ i64 │
╞══════╡
│ 3 │
│ 4 │
└──────┘
df.columns — это список имен столбцов, вы можете проиндексировать его, чтобы получить последнее имя столбца.
df.get_column(df.columns[-1])
get_columns() здесь полезен.
import polars as pl
df = pl.DataFrame({"foo": [1, 2, 3], "bar": [4, 5, 6]})
df.get_columns()[-1]
shape: (3,)
Series: 'bar' [i64]
[
4
5
6
]
Также есть pl.first() / pl.last() например.
df.select(pl.last())
shape: (2, 1)
┌──────┐
│ col3 │
│ --- │
│ i64 │
╞══════╡
│ 3 │
│ 4 │
└──────┘