如何根據數據值對熊貓數據框中的列進行分類？

Question

我有一個數據框，其中包含一個包含實際值的列。

我想有一個額外的列，根據繼承人的大小分類這些值。例如，我想知道一個值是否屬於一組最大值中最小值的組。我希望這兩個團體擁有相同數量的元素。

例如。如果我有下列值：

[1,2,3,4,40,50] 

我想映射1,2和3比1和4，40，和50至2有一種簡單的方法來做到這在數據幀。

在上面的例子中，我只使用了兩組。但我想保持靈活性。例如，對於三組我要地圖1和2比1，3和4 2，40和50至3

Answer 1

import heapq 
import random 
x = range(100000) 
random.shuffle(x) 
print(heapq.nlargest(2, x)) 

給出：[99999，99998]

現在只需做一些像：

max_column = heapq.nlargest(len(x)/2, x) 

這應該給你一個「大」的列表的一半，併爲小堆做同樣的事情。