我對串列中的分組元素有以下問題。在數字影像轉換之后,我將孔中心分開并將它們收集在值串列中,然后在相鄰元素之間的計算差異之后我得到了diff_ar。現在我想獲取屬于一個組/集群的元素的索引。我假設一個部分中元素之間的最大差異應該小于 3。此外,只有在內部至少有 7 個元素時才能創建組。因此,我期望包含索引開始和索引結束的每個 od 檢測組(在本例中為 2)的元組串列。
更好的問題陳述的影像
values = [73.0, 143.0, 323.0, 324.0, 325.0, 325.0, 325.0, 325.0, 325.5,
325.5, 326.0, 326.0, 326.0, 326.0, 406.0, 406.5, 432.5, 433.0,
433.5, 434.5, 435.0, 435.0, 436.0, 436.5, 437.5, 438.0]
diff_ar = [70.0, 180.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.5, 0.0, 0.5, 0.0,
0.0, 0.0, 80.0, 0.5, 26.0, 0.5, 0.5, 1.0, 0.5, 0.0, 1.0, 0.5,
1.0, 0.5]
expected_output = [(2,12),(16,24)]
uj5u.com熱心網友回復:
第一個解決方案:使用more_itertools.split_when.
import more_itertools
values = [73.0, 143.0, 323.0, 324.0, 325.0, 325.0, 325.0, 325.0, 325.5,
325.5, 326.0, 326.0, 326.0, 326.0, 406.0, 406.5, 432.5, 433.0,
433.5, 434.5, 435.0, 435.0, 436.0, 436.5, 437.5, 438.0]
threshold = 3
min_items = 7
groups = [(g[0][0], g[-1][0]) for g in more_itertools.split_when(enumerate(values), lambda x,y: abs(x[1]-y[1])>=threshold) if len(g) >= min_items]
print(groups)
# [(2, 13), (16, 25)]
第二種解決方案:自己撰寫回圈。
def split_values(values, threshold, min_items):
result = []
prev = values[0]
last_cut = 0
for i,x in enumerate(values[1:], start=1):
if abs(x - prev) >= threshold:
if i - last_cut >= min_items:
result.append((last_cut, i-1))
last_cut = i
prev = x
if len(values) - last_cut >= min_items:
result.append((last_cut, len(values)-1))
return result
values = [73.0, 143.0, 323.0, 324.0, 325.0, 325.0, 325.0, 325.0, 325.5,
325.5, 326.0, 326.0, 326.0, 326.0, 406.0, 406.5, 432.5, 433.0,
433.5, 434.5, 435.0, 435.0, 436.0, 436.5, 437.5, 438.0]
print(split_values(values, 3, 7))
# [(2, 13), (16, 25)]
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/322323.html
