Почему совокупная сумма по каждой группе дает все 1?Python

Программы на Python
Ответить
Anonymous
 Почему совокупная сумма по каждой группе дает все 1?

Сообщение Anonymous »

В Pandas я использую .groupby().cumsum() для создания столбца счетчика:

Код: Выделить всё

import pandas as pd

df = pd.DataFrame({'ID':['A','B','A','A','B','B','C','D','D','C']})

df['count'] = df['ID'].ne(df['ID'].shift()).groupby(df['ID']).cumsum()

Код: Выделить всё

  ID  count
0  A      1
1  B      1
2  A      2
3  A      2
4  B      2
5  B      2
6  C      1
7  D      1
8  D      1
9  C      2
При использовании того же подхода в Polars я вместо этого получаю все 1. Почему?

Код: Выделить всё

import polars as pl

df = pl.DataFrame({'ID':['A','B','A','A','B','B','C','D','D','C']})

df.with_columns(
count = (pl.col('ID').ne(pl.col('ID').shift()).fill_null(True).cum_sum()).over('ID')
)

Код: Выделить всё

┌─────┬───────┐
│ ID  ┆ count │
│ --- ┆ ---   │
│ str ┆ u32   │
╞═════╪═══════╡
│ A   ┆ 1     │
│ B   ┆ 1     │
│ A   ┆ 1     │
│ A   ┆ 1     │
│ B   ┆ 1     │
│ B   ┆ 1     │
│ C   ┆ 1     │
│ D   ┆ 1     │
│ D   ┆ 1     │
│ C   ┆ 1     │
└─────┴───────┘
Как создать тот же столбец счетчика, что и в решении pandas?

Подробнее здесь: https://stackoverflow.com/questions/798 ... -in-all-1s
Ответить

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

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

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

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

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