基于列值创建具有重复行的数据
基于列值创建具有重复行的数据
提问于 2022-05-09 16:11:10
我正在尝试展开一个有两个列的数据集，并在python中展开它。
Basket | Times
______________|_______
Bread | 5
Orange, Bread | 3 我想，根据“泰晤士报”专栏中的数字，有那么多行。所以对于上面的例子
Newcolumn
_______
Bread1
Bread2
Bread3
Bread4
Bread5
Orange, Bread1
Orange, Bread2
Orange, Bread3 回答 2
Stack Overflow用户
回答已采纳
发布于 2022-05-09 16:35:18
您可以尝试在行上使用apply生成所需的列表，并对列进行explode
df['Newcolumn'] = df.apply(lambda row: [f"{row['Basket']}_{i+1}" for i in range(row['Times'])], axis=1)
df = df.explode('Newcolumn', ignore_index=True)print(df)
Basket Times Newcolumn
0 Bread 5 Bread_1
1 Bread 5 Bread_2
2 Bread 5 Bread_3
3 Bread 5 Bread_4
4 Bread 5 Bread_5
5 Orange, Bread 3 Orange, Bread_1
6 Orange, Bread 3 Orange, Bread_2
7 Orange, Bread 3 Orange, Bread_3Stack Overflow用户
发布于 2022-05-09 16:23:59
使用np.repeat重复每个值所需的次数。然后groupby和cumcount添加所需的后缀：
import numpy as np
srs = np.repeat(df["Basket"],df["Times"])
output = (srs+srs.groupby(level=0).cumcount().add(1).astype(str)).reset_index(drop=True)
>>> output
0 Bread1
1 Bread2
2 Bread3
3 Bread4
4 Bread5
5 Orange, Bread1
6 Orange, Bread2
7 Orange, Bread3
dtype: object页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/72175073
复制相关文章
相似问题
腾讯云开发者
