超越 value_counts() [关闭]

Question

我有一个 pandas 数据框，每个数据框包括一些 values 和 count：

df = pd.DataFrame({'value':[1,2,3,11,12,13,21,22,23], 'count':[100,200,300, 1100,1200,1300, 2100,2200,2300]})

value count
1     100
2     200
3     300
11    1100
12    1200
13    1300
21    2100
22    2200
23    2300

我想要的是在value 的给定相等大小的窗口中添加计数。例如，对于 window_size=10，我想要：

value     count
[0,  10)  600
[10, 20)  3600
[20, 30)  6600

最好的方法是什么？

Answer 1

您可以使用pd.cut 将列value 分类为基于窗口大小的离散区间，然后使用此分箱列按列count 分组，并使用sum 聚合：

w = 10
g = pd.cut(df['value'], np.r_[0:df['value'].max() + w : w], right=False)
s = df['count'].groupby(g).sum()

---

value
[0, 10)      600
[10, 20)    3600
[20, 30)    6600
Name: count, dtype: int64