Pandas使用用户定义的区间进行分组切割

Question

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

我已经尝试了许多解决方案，但都没有成功 - 欢迎提出任何建议！

- wrcobb

Answer 1

Something like this?

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)

- chrisb