Как включить первый соответствующий шаблон в виде столбцаPython

Программы на Python
Ответить
Anonymous
 Как включить первый соответствующий шаблон в виде столбца

Сообщение Anonymous »

У меня есть фрейм данных df.
>>> import polars as pl
>>>
>>>
>>> df = pl.DataFrame({"col": ["row1", "row2", "row3"]})
>>> df
shape: (3, 1)
┌──────┐
│ col │
│ --- │
│ str │
╞══════╡
│ row1 │
│ row2 │
│ row3 │
└──────┘

Теперь я хочу создать новый столбец. Это должен быть первый совпавший шаблон в столбце.
Например, для шаблона 1|2 должен выдаваться следующий результат.
┌──────┬───────┐
│ col ┆ new │
│ --- ┆ --- │
│ str ┆ str │
╞══════╪═══════╡
│ row1 ┆ 1 │
│ row2 ┆ 2 │
│ row3 ┆ null │
└──────┴───────┘

Я пробовал использовать API выражений, но он возвращает логические значения.
>>> df.with_columns(new=pl.col('col').str.contains("1|2"))
shape: (3, 2)
┌──────┬───────┐
│ col ┆ new │
│ --- ┆ --- │
│ str ┆ bool │
╞══════╪═══════╡
│ row1 ┆ true │
│ row2 ┆ true │
│ row3 ┆ false │
└──────┴───────┘


Подробнее здесь: https://stackoverflow.com/questions/795 ... s-a-column
Ответить

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

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

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

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

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