list内のtuple内の要素を元にして、listを処理したい

Question

お世話になっております。ややこしいタイトルで申し訳ありません。
今回お伺いしたいのは、pythonのリストに内包しているtupleの要素を元にリストの処理を行えるかについてです。

以下のようなtupleを内包したlistがあります。

list = [
('1aa', '1bb', '1cc', 123,),
('1aa', '1bb', '4cc', 012,),
('1aa', '3bb', '3cc', 789,),
('2aa', '2bb', '2cc', 123,),
('2aa', '2bb', '2cc', 456,),
('3aa', '3bb', '3cc', 456,),
]

このlistに対し、

• list[x][0]及びlist[x][1]が重複しているものしかない場合にlistから除外

上記リストだと、list[x][0]が'1aa'のものは3つあり、内2要素が
'1aa', '2bb'であるが、3要素目が'1aa', '3bb'なので除外しない
逆に、list[x][0]が'2aa'のものは'2aa', '2bb'と同じものしかない為除外

• そもそもリスト内に一つしかないtupleをlistから除外

逆に、list[x][0]が'3aa'のものは1つしか無いので除外

このような除外処理って可能なのでしょうか、
自分でも考えてはみたのですが、Pythonを始めて触ったレベルの為どうにも難航しております。
お知恵拝借出来ればと思います。宜しくお願い致します。

Answer 1

これ、条件を2つ書いていますが実質同じですよね。

list[x][0]及びlist[x][1]が重複しているものしかない場合にlistから除外

解釈: タプル0の値でグルーピングして、タプル1の値が1種類なら、除外

そもそもリスト内に一つしかないtupleをlistから除外
逆に、list[x][0]が'3aa'のものは1つしか無いので除外

解釈: タプル0の値でグルーピングして、タプル1の値が1種類なら、除外

この解釈でよければ以下のように書けます。

>>> data = [  # listだとlist関数を隠蔽するのでdataにします
... ('1aa', '1bb', '1cc', 123,),
... ('1aa', '1bb', '4cc', 012,),
... ('1aa', '3bb', '3cc', 789,),
... ('2aa', '2bb', '2cc', 123,),
... ('2aa', '2bb', '2cc', 456,),
... ('3aa', '3bb', '3cc', 456,),
... ]
...

>>> # 除外条件を求めるために、タプル0でタプル1をグルーピング
>>> conditions = {}
>>> for d in data:
...     if d[0] not in conditions:
...         conditions[d[0]] = set()
...     conditions[d[0]].add(d[1])
... 
>>> conditions
{'3aa': set(['3bb']), '1aa': set(['3bb', '1bb']), '2aa': set(['2bb'])}

>>> # タプル1が1つしかない行をタプル0で特定して削除
>>> for key in conditions:
...     if len(conditions[key]) == 1:
...         data = [x for x in data if x[0] != key]

>>> # 結果
>>> data
[('1aa', '1bb', '1cc', 123), ('1aa', '1bb', '4cc', 10), ('1aa', '3bb', '3cc', 789)]

---

（おまけ）これを短く書きなおしてみます。

>>> data = [  # listだとlist関数を隠蔽するのでdataにします
... ('1aa', '1bb', '1cc', 123,),
... ('1aa', '1bb', '4cc', 012,),
... ('1aa', '3bb', '3cc', 789,),
... ('2aa', '2bb', '2cc', 123,),
... ('2aa', '2bb', '2cc', 456,),
... ('3aa', '3bb', '3cc', 456,),
... ]
...
>>> for k, d in groupby(data, itemgetter(0)):
...     if len(set(itemgetter(1)(x) for x in d)) == 1:
...         data = [x for x in data if x[0] != k]
...
>>> data
[('1aa', '1bb', '1cc', 123), ('1aa', '1bb', '4cc', 10), ('1aa', '3bb', '3cc', 789)]

Comments

[dohq]

ありがとうございます!
無事期待どおりの結果を得る事が出来ました!
グルーピングという考えがまったく無かった為、目から鱗の思いでした。
ありがとうございました!

Answer 2

条件文を整理すると

「１つ目の要素が等しい」かつ「２つ目の要素が異なる」tupleが他に存在しない場合は除外する

となるので、こんな感じで書けると思います。

data = [
('1aa', '1bb', '1cc', 123,),
('1aa', '1bb', '4cc',  12,),
('1aa', '3bb', '3cc', 789,),
('2aa', '2bb', '2cc', 123,),
('2aa', '2bb', '2cc', 456,),
('3aa', '3bb', '3cc', 456,),
]

res = []
for d0 in data:
    if any([d0[0] == d1[0] and d0[1] != d1[1] for d1 in data]):
        res.append(d0)
print(res)
# => [('1aa', '1bb', '1cc', 123), ('1aa', '1bb', '4cc', 12), ('1aa', '3bb', '3cc', 789)]

ワンライナーで書くとこんな感じ（蛇足）

res = [d for d in data if any([d[0] == d1[0] and d[1] != d1[1] for d1 in data])]

Comments

[dohq]

ありがとうございます!
any関数を知らなかった為、そのような方法があったのかと勉強になります。
まだまだpython独特の関数には驚かされそうです。
解答頂いたお二方ともにベストアンサーを差し上げたいのですが…
なにはともあれ、ありがとうございました!