划分数量以便每行得到一个数量- python

0yg35tkg 于 2022-12-02 发布在 Python

关注(0)|答案(1)|浏览(150)

我有一个有数量和价格的数据框。我想得到一个价格与数量相同但每行只有一个数量的数据框。
数据框架：

Name | qty | Price
Apple | 3 | 3.50
Avocado | 2 | 1.50

预期输出：

Name | qty| Price
Apple | 1 | 3.50
Apple | 1 | 3.50
Apple | 1 | 3.50
Avocado | 1 | 1.50
Avocado | 1 | 1.50

老实说，我不知道如何用Python的方式编写这个代码。

1条答案

我们可以使用df.index.repeat，然后将所有行的qty设置为1。

df = df.loc[df.index.repeat(df['qty'])].reset_index(drop=True)
df['qty'] = 1

输出量：

Name  qty  Price
0    Apple    1    3.5
1    Apple    1    3.5
2    Apple    1    3.5
3  Avocado    1    1.5
4  Avocado    1    1.5