熊猫用用户定义的箱子切割

Question

我正在使用Python并在Pandas中使用剪切功能。我想让我的pd.cut中的bin基于用户定义的逗号分隔整数，具有预定义的顶部和底部边界。换句话说，我想将bins预定义为[0, 100000]，然后将用户定义的整数附加到这组数字。如果用户输入5, 100, 5000，则bins将变为[0, 5, 100, 5000, 100000]，然后将与pd.cut(df['Quantity'], bins一起使用。

我已经尝试了许多解决方案，但到目前为止还无济于事 - 任何输入都将不胜感激！

Answer 1

这样的东西？

bin_range = [0, 100000]
bin_input = raw_input("Enter bins as comma seperated list")
bin_input_parsed = [int(x.strip()) for x in bin_input.split(',')]

bins = [bin_range[0]] + bin_input_parsed + [bin_range[1]]
pd.cut(df['Quantity'], bins)