Лучший способ создать столбец, содержащий последовательные значения ⇐ Python

[Anonymous]

У меня есть следующий фрейм данных:

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

df = pd.DataFrame({'sku' : ['A','A','A','A', 'B', 'B', 'C','C','C'],
'price': [10,10,10,10, 8,8,9,9,9]})

out:

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

    sku price
0   A   10
1   A   10
2   A   10
3   A   10
4   B   8
5   B   8
6   C   9
7   C   9
8   C   9

Моя проблема
Я хочу создать новый столбец (с именем t), содержащий последовательные значения в соответствии в номер столбца, чтобы получить следующий результат:

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

 sku    price   t
0   A   10  1
1   A   10  2
2   A   10  3
3   A   10  4
4   B   8   1
5   B   8   2
6   C   9   1
7   C   9   2
8   C   9   3

Мой текущий код:

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

t = [1]
for i in range(1, len(df)):
if df['sku'][i-1] != df['sku'][i]:
t.append(1)
else:
t.append(t[i-1] + 1)

df['t'] = t

Есть ли лучший способ сделать это?


Подробнее здесь: https://stackoverflow.com/questions/793 ... ial-values