熊猫分类发现价值箱

Question

我想离散化数组：我正在使用的是pandas.cut

import numpy as np
import pandas as pd
arr = np.array([1, 2, 3, 4, 5, 6 , 7])
factor = pd.cut(arr, 2)

如何获得任意值的因子仓（即第一个仓中为2.5）？
还是有更好的解决方案（当然，我总是可以自己做for循环...）？

Answer 1

保存垃圾箱并再次使用pd.cut：

factor, bins = pd.cut(arr, 2, retbins=True)
pd.cut([2.5], bins=bins)
Out: 
[(0.994, 4]]
Categories (2, object): [(0.994, 4] < (4, 7]]