Apache Spark從邊界列表中獲取數字

Question

我有一個進入Spark函數的數字列表。每個數字是一個類別的下限...... E.g，

6.252 1 
5.502 2 
4.387 3 

然後我輸入一個數字到這個函數。基本上我想看看它列入哪個列表的索引。所以8.5將會屬於第一類，6.1將在第二類。我嘗試使用絕對值使用

myIndex = list.index(min(list, key=lambda x:abs(x-inputVal)))) 

但這可能會給我錯誤的桶，因爲它只是尋找接近。我還想過通過列表遍歷蠻力，並抓取當前值和下一個值，但必須有一個更優雅的解決方案。

Answer 1

您可以使用對開模塊：

import bisect 

breaks = sorted([4.387, 5.502, 6.252]) 

len(breaks) - bisect.bisect_left(breaks, 4.3)