Я пытаюсь создать 2 новых столбца в выводе при проверке условия, но не знаю, как это сделать.
пример df: сильный>
so_df = pl.DataFrame({"low_limit": [1, 3, 0], "high_limit": [3, 4, 2], "value": [0, 5, 1]})
low_limit high_limit value
i64 i64 i64
1 3 0
3 4 5
0 2 1
Работающий код для создания одного столбца:
so_df.with_columns(pl.when(pl.col('value') > pl.col('high_limit'))
.then(pl.lit("High"))
.when((pl.col('value') < pl.col('low_limit')))
.then(pl.lit("Low"))
.otherwise(pl.lit("Within Range")).alias('Flag')
)
вывод
low_limit high_limit value Flag
i64 i64 i64 str
1 3 0 "Low"
3 4 5 "High"
0 2 1 "Within Range"
Проблема/сомнение: Создание двух столбцов не работает
so_df.with_columns(pl.when(pl.col('value') > pl.col('high_limit'))
.then(Flag = pl.lit("High"), Normality = pl.lit("Abnormal"))
.when((pl.col('value') < pl.col('low_limit')))
.then(Flag = pl.lit("Low"), Normality = pl.lit("Abnormal"))
.otherwise(Flag = pl.lit("Within Range"), Normality = pl.lit("Normal"))
)
Желаемый результат:
low_limit high_limit value Flag Normality
i64 i64 i64 str str
1 3 0 "Low" "Abnormal"
3 4 5 "High" "Abnormal"
0 2 1 "Within Range" "Normal"
Я знаю, что могу сделать еще один with_Columns и снова использовать When-then, но это потребует двойных вычислений. Итак, как мне создать 2 новых столбца за один раз?
что-то вроде:
if (condition):
Flag = '',
Normality = ''
Подробнее здесь: https://stackoverflow.com/questions/790 ... -in-polars
Как создать несколько столбцов в выводе при условии в Polars? ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение
-
-
Кадр данных Polars: добавление столбцов при условии, что другой столбец дает разную длину
Anonymous » » в форуме Python - 0 Ответы
- 19 Просмотры
-
Последнее сообщение Anonymous
-