pythonのlistの操作

Question

今、pythonの勉強をしている学生です。
どうしてもわからない問題があるので回答よろしくお願いします。
下記のようなlistの例があった時、①②のような操作を行いたいです。

①()内の左の要素が連続する場合後ろの要素を抜き出したい
ー＞例えば,(33, 695), (34, 685)の場合、(34,685)を抽出したいということです
②前後ともに連続しない（孤立している）場合も、抜き出したい
ー＞例えば、listの例では(132, 737)がそれに該当します。

理想的な回答としては
[(34, 685), (132, 737), (229, 674), (328, 704)]
となることです。

listの例
[(33, 695), (34, 685), (132, 737), (228, 688), (229, 674), (327, 684), (328, 704)]

今は下記プログラムのように連続した場合、後ろを抽出するプログラムを何とか書けたところです。②で詰まっています。
非常に汚いコードで申し訳ないです。
lの長さは本来もっと長いことを想定しています。

cnt = 0
k = 0
a = []
final = []
l = [(33, 695), (34, 685), (132, 737), (228, 688), (229, 674), (327, 684), (328, 704)]

for i in l:
    if cnt == 0:
        a.append(i)
    else:
        if i[0] - a[k][0] < 2:
            final.append(i)
            a.append(i)
        else:
            a.append(i)
        k += 1
    cnt += 1

[(34, 685), (229, 674), (328, 704)]

宜しくお願い致します。

環境
Windows 10
python 3.7

Answer 1

素晴らしい解決方法をありがとうございました

Answer 2

わかりやすい解答ありがとうございます

Answer 3

前の項の値との差の list を作り、それをつかって残す項を調べるjという方法でかいてみました。
いろいろな場合の例のテストもしてみています。

def runlength_tail(l):
    if not l or len(l) == 0:
        return l

    l_keys = [x[0] for x in l]
    diffs = [0] + [l_keys[i + 1] - l_keys[i] for i in range(0, len(l_keys) - 1)] + [0]
    # print(diffs)
    return [l[i] for i in range(0, len(diffs) - 1)  if (diffs[i + 1] != 1)]

l = [
    (33, 695), (34, 685), 
    (132, 737), 
    (228, 688), (229, 674),
    (327, 684), (328, 704)
]
print(runlength_tail(l))

l = None
print(l, " -> ", runlength_tail(l))
l = [(1, 1)]
print(l, " -> ", runlength_tail(l))
l = [(1, 1), (2, 2)]
print(l, " -> ", runlength_tail(l))
l = [(1, 1), (2, 2), (3, 3)]
print(l, " -> ", runlength_tail(l))
l = [(1, 1), (3, 3), (5, 5)]
print(l, " -> ", runlength_tail(l))
l = [(1, 1), (2, 2), (5, 5)]
print(l, " -> ", runlength_tail(l))
l = [(1, 1), (4, 4), (5, 5)]
print(l, " -> ", runlength_tail(l))
l = [(1, 1), (2, 2), (3, 3),
     (5, 5), (6, 7), (7, 7),
     (9, 9), (10, 10), (11, 11)]
print(l, " -> ", runlength_tail(l))
l = [(1, 1), (2, 2), (3, 3),
     (5, 5), (6, 7), (7, 7),
     (9, 9), 
     (11, 11), (12, 12), (13, 13)]
print(l, " -> ", runlength_tail(l))

実行例

Answer 4

以下のようなアプローチではどうでしょうか。

リストから連続する3つの要素を取り出して、左側から順に left, center, right と名前を付けます。
条件①は、left[0] + 1 == center[0] のときcenterを抜き出す
条件②は、left[0] + 1 != center[0] かつ center[0] +1 != right[0] のときcenterを抜き出す
として処理します。
3つの要素を取り出す窓を1つずつ後ろへずらしていけば、リストの最後まで処理できます。

l = [(32, 123), (33, 695), (34, 685), (132, 737), (228, 688), (229, 674),
     (327, 684), (328, 704), (500, 5), (699, 6), (700, 7)]

res = []
for left, center, right in zip(l, l[1:], l[2:]+[l[-1]]):
    if (left[0]+1 == center[0]) or (left[0]+1 != center[0] and center[0]+1 != right[0]):
        res.append(center)
print(res) # [(33, 695), (34, 685), (132, 737), (229, 674), (328, 704), (500, 5), (700, 7)]

Answer 5

逆から処理すれば簡単に書けます。

l = [(33, 695), (34, 685), (132, 737), (228, 688),
    (229, 674), (327, 684), (328, 704)]

final = []
k = None
for i in reversed(l):
    if i[0]+1 != k:
        final.insert(0,i)
    k = i[0]

print(final)

元のコードのaの位置づけが分からないので無視しました。

と思ったけど、lが非常に長いということであれば、逆順に並べ替えるコストが気になるので、添え字を使うのでしょうか。

for i in range(len(l)-1, -1, -1):
    if l[i][0]+1 != k:
        final.insert(0,l[i])
    k = l[i][0]

Comments

[otn]

そうか、添え字使ったら、前からやるのとあまり変わらないですね。

[otn]

こういうのは、Rubyが得意です。
l = [[33, 695], [34, 685], [132, 737], [228, 688],
[229, 674], [327, 684], [328, 704]]
final = l.chunk_while{|x,y| x[0]+1==y[0]}.map(&:last)
p final

Answer 6

lst = [(33, 695), (34, 685), (132, 737), (228, 688), (229, 674), (327, 684), (328, 704)]
# 念のため最初の要素でソートします。
lst = sorted(lst, key=lambda x: x[0])
lstN = [s for i, s in enumerate(lst) if i == len(lst)-1 or s[0]+1 != lst[i+1][0]]
print(lstN)  # [(34, 685), (132, 737), (229, 674), (328, 704)]

Answer 7

外部のコードサイトで書いたのでお手数ですがそちら見てください
repl.it

こんな感じでいいんでしょうか