Как создать несколько столбцов в выводе при условии в Polars?Python

Программы на Python
Ответить Пред. темаСлед. тема
Anonymous
 Как создать несколько столбцов в выводе при условии в Polars?

Сообщение Anonymous »

Я пытаюсь создать 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
Реклама
Ответить Пред. темаСлед. тема

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

  • Похожие темы
    Ответы
    Просмотры
    Последнее сообщение

Вернуться в «Python»