pythonの二次元配列のデータで前の行との同値判定をおこないたい場合のリスト内包表記の書き方

Question

以下のような二次元配列のデータでa, b, c ごとに処理を行わせたいです。例えば、一つ前の行が同値である場合だけfor文を回し、特定の関数の引数にいれ、aからbになった場合には、また別の処理にかけたいです。

なにか、お気づきの点ありがましたらご教示いただけませんでしょうか？

data = [['a', 'B3', 0], ['a', 'B3', 0], ['a', 'B3', 0], ['a', 'B2', 0], ['a', 'B2', 0], ['b', 'B2', 0], ['b', 'B2', 1], ['b', 'B2', 0], ['b', 'B2', 0], ['b', 'B2', 1], ['c', 'B2', 0], ['c', 'B2', 0], ['c', 'B2', 1]]

＊pandusは、利用しない想定です。

Answer 1

itertools.groupbyを使用する案はどうでしょうか。
以下のようなイメージで、リストの先頭部分('a', 'b', 'c')で同じ値が連続するものをグループ化できます。

from itertools import groupby

data = [['a', 'B3', 0], ['a', 'B3', 0], ['a', 'B3', 0], ['a', 'B2', 0], ['a', 'B2', 0], ['b', 'B2', 0], ['b', 'B2', 1], ['b', 'B2', 0], ['b', 'B2', 0], ['b', 'B2', 1], ['c', 'B2', 0], ['c', 'B2', 0], ['c', 'B2', 1]]
ans = [list(v) for _, v in groupby(data, key=lambda x: x[0])]

for a in ans:
    print(a)
# [['a', 'B3', 0], ['a', 'B3', 0], ['a', 'B3', 0], ['a', 'B2', 0], ['a', 'B2', 0]]
# [['b', 'B2', 0], ['b', 'B2', 1], ['b', 'B2', 0], ['b', 'B2', 0], ['b', 'B2', 1]]
# [['c', 'B2', 0], ['c', 'B2', 0], ['c', 'B2', 1]]

Comments

[sequelanonymous]

ありがとうございます。私の方では、こうかいてみました。どちらのほうがいいのかいまいちわからず。。

ans = [[row for row in data if row[0] == char] for char in list({row[0] for row in data})]

Answer 2

list内包表記とset内包表記の組み合わせにより重複を排除して二次元配列を作成することで解決

ans = [[row for row in data if row[0] == char] for char in list({row[0] for row in data})]

Answer 3

こういうことですか？

temp = []

for i in data:
    if temp == i:
        print('A', i)
    else:
        print('B', i)

    temp = i