python - 如何从数据框中创建一个列表，其中包含字符串列的每个值乘以第二列？

Question

这个问题已经有答案了:

Repeat rows in a pandas DataFrame based on column value (4 个回答)

已关闭 3 年前。

提供示例后，问题会变得更加清晰。我有一个包含两列的数据框，一列为字符串类型，一列为整数类型:

Col1   Col2
-------------
str1    2
str2    4
str3    1

现在我需要一个列表，其中包含 Col1 中的字符串乘以 Col2 中的数字，即 ['str1', 'str1', 'str2', 'str2', 'str2', 'str2', 'str3'] .

最有效的方法是什么？

Answer 1

import numpy as np
np.repeat(df.Col1.values, df.Col2.values)

输出:

array(['str1', 'str1', 'str2', 'str2', 'str2', 'str2', 'str3'])