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

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)