Развернуть/развернуть DataFrame, содержащий данные подсчета, в одну строку для каждого элемента ⇐ Python

[Anonymous]

Контекст: преобразование данных для задачи логистической регрессии. У меня есть следующая структура данных:
df = pd.DataFrame({"group": ["A", "B"], "total": [3, 5], "occurrence": [2, 1]})

Я хочу кое-что сделать. например pd.explode, но создает одну строку для элемента итога, т. е. 5+6 строк, где количество строк содержит 1, а остальные 0 (либо в столбцах вхождения, либо в новом целевом столбце).
В настоящее время я делаю это итеративно, что довольно медленно для больших данных:
expanded = []
for ix, row in df.iterrows():
for i in range(row["total"]):
row["y"] = 1 if i < row["occurrence"] else 0
expanded.append(row.copy())
df_out = pd.DataFrame(expanded).reset_index(drop=True)
df_out.drop(["total", "occurrence"], axis=1, inplace=True)
df_out

group y
0 A 1
1 A 1
2 A 0
3 B 1
4 B 0
5 B 0
6 B 0
7 B 0



Подробнее здесь: https://stackoverflow.com/questions/791 ... w-per-item