从值和值计数的数据集构造一个列表

Question

要将数据Score扩展到基于Count的Scores列表中，pandas和numpy中是否有比下面更好的方法？你知道吗

import pandas as pd
import numpy as np

data = {
    "Count": [1, 3, 2],
    "Score": [2, 5, 8]
}

df = pd.DataFrame(data)

scores = []

for c, s in zip(df['Count'], df['Score']):
    for i in range(0, c):
        scores.append(s)

print(scores)

预期产量：

[2, 5, 5, 5, 8, 8]

Answer 1

IIUC，您可以使用^{}：

df['Score'].repeat(df['Count']).tolist()

或^{}：

np.repeat(df['Score'],df['Count']).tolist()

或^{}：

df.loc[df.index.repeat(df['Count']),'Score'].tolist()

---

[2, 5, 5, 5, 8, 8]