是否有 python 或 pandas function 将列表切割成大小相等或尽可能对称的组？

Question

我想要一个功能类似于pandas.qcut但是，它给了我尽可能“平衡”或“对称”的分裂。

目前，如果我使用：

pd.qcut(range(1, 11), 3, labels=False, duplicates="drop")

我得到：

array([0, 0, 0, 0, 1, 1, 1, 2, 2, 2], dtype=int64)

但我希望中间组有四个条目，即：

array([0, 0, 0, 1, 1, 1, 1, 2, 2, 2], dtype=int64).

Answer 1

Integer 算术可以做到。

import pandas as pd
pd.Series([i%3 for i in range(1, 11)]).sort_values().values

output

array([0, 0, 0, 1, 1, 1, 1, 2, 2, 2])