python - Pandas 切割无限上/下界

Question

pandas cut() documentation声明:“超出范围的值在生成的分类对象中将为 NA。”当上限不一定明确或不重要时，这就变得很困难。例如:

cut (weight, bins=[10,50,100,200])

将产生垃圾箱:

[(10, 50] < (50, 100] < (100, 200]]

所以 cut (250, bins=[10,50,100,200])将产生 NaN ，同样会 cut (5, bins=[10,50,100,200]) .我想做的是产生类似 > 200 的东西对于第一个例子和< 10第二个。

我意识到我可以做到 cut (weight, bins=[float("inf"),10,50,100,200,float("inf")])或等效的，但我遵循的报告样式不允许 (200, inf] 之类的东西.我也意识到我实际上可以通过 labels 指定自定义标签cut() 上的参数，但这意味着每次调整时都要记得调整它们 bins ，这可能经常发生。

我是否已经穷尽了所有的可能性，还是 cut() 中有什么东西？或 pandas 中的其他地方那会帮助我做到这一点？我正在考虑为 cut() 编写一个包装函数这会自动从垃圾箱中生成所需格式的标签，但我想先在这里检查一下。

Answer 1

您可以使用 float("inf") 作为 bins 列表的上限，使用 -float("inf") 作为下限。它将删除 NaN 值。