Pandas:value_counts函数的反转

时间:2016-02-18 21:19:41

标签: python numpy pandas

生成value_counts函数的反向输出的快速方法是什么?

例如,如果我有以下系列:

1      24
2       2
3       1
4       2
5       3
6      12
7      21
8     204
9     400
10     71
11    160
Name: foo, dtype: float64

如何简明地生成以下数组?

numpy.array([1, 1, 1, ... , 2, 2, 3, 4, 4, 5, 5, 5, 6, ... ])

1 个答案:

答案 0 :(得分:4)

您可以使用np.repeat。如果您的系列名为s,则可以写:

np.repeat(s.index.values, s.values)

此处s.index.values是要重复的值,s.values指定每个值应重复的次数。输出是一维数组。