如何根據同一 pandas dataframe 中的列的值復制行

Question

我需要做一些行復制。 我有一個看起來像這樣的 dataframe：

label	數數
HA1	5
HA2	3
HA3	1
HA4	4

我要做的是通過“計數”列中的數字復制每一行。 所以 output 應該是這樣的：

老實說，我不是編碼員，因此我試圖在更大的數據集上執行此操作。 希望能從這個社區得到一些想法。

謝謝。

-大耳朵

Answer 1

嘗試reindex + repeat

out = df.reindex(df.index.repeat(df['count']))

Answer 2

使用 df.values 的替代方法

import pandas as pd
out = pd.DataFrame(df.values.repeat(df['count'], axis=0), columns=df.columns)