ある数値以下の間隔の場合にリストを結合したい

Question

実現したいこと

以下のような2次元リストがあります。各要素の１つ目は開始秒数、２つ目は終了秒数を表しているものとします。

[[0, 3], [3.2, 5], [5.3, 7], [8, 9], [9.3, 10]]

このとき、ある終了秒数から次の開始秒数が0.5以下の場合に、リストを結合したいです。
例えば上記の例ですと、

[[0, 7], [8, 10]]

という出力が期待する結果になります。
必ず開始秒数と終了秒数はペアで存在しますが、リストの個数は不定です。

上記を実現するにはどのようなコードになるか教えて頂けないでしょうか。

Answer 1

list2に結果を格納します。以下のようにしてください・

list1 = [[0, 3], [3.2, 5], [5.3, 7], [8, 9], [9.3, 10]]
list2 = []
list2.append(list1[0])
for x in list1[1:]:
    if x[0] - list2[-1][1] > 0.5:
        list2.append(x)
    else:
        list2[-1][1] = x[1]
print(list2)

実行結果
[[0, 7], [8, 10]]

Answer 2

ChatGPTの回答

以下はPythonでリストを結合するためのコード例です。

intervals = [[0, 3], [3.2, 5], [5.3, 7], [8, 9], [9.3, 10]]
merged_intervals = []
start = intervals[0][0]
end = intervals[0][1]

for interval in intervals[1:]:
    if interval[0] - end <= 0.5:
        end = interval[1]
    else:
        merged_intervals.append([start, end])
        start = interval[0]
        end = interval[1]

merged_intervals.append([start, end])
print(merged_intervals)

上記のコードでは、まず最初の区間を取り出して開始秒数と終了秒数を変数startとendに格納します。その後、forループでリストの要素を1つずつ処理し、現在の区間の開始秒数と前の区間の終了秒数の差が0.5以下の場合はリストを結合します。差が0.5以上の場合は現在の区間を新しい区間としてmerged_intervalsリストに追加し、startとendを更新します。最後に、最後の区間をmerged_intervalsリストに追加しています。