pythonで１つのリストから重複している要素を削除し，それを他のリストに反映させる。

Question

#やりたいこと
まず，pythonの1つのリストから，順序を維持したまま重複している要素を削除します。
その後，他のリストにて，削除した位置と同じ位置の要素を削除したいです。

#実行例
具体的な実行例を以下に示します。
ここではリストaとリストbの2つのリストがあり，中身は以下の通りです。
a=[2,1,2,3]
b=[a,b,c,d]

この場合に，リストaで重複している要素を順序を維持したまま削除し，リストaで削除した要素の位置と同じ位置のリストbの要素を削除したいです。
削除後は以下のようになります。
a=[2,1,3]
b=[a,b,d]

削除後の両リストを見ると，リストaの重複していた2が削除され，リストbでは，同じ位置の要素'c'が消えています。
pythonでのこの処理を行うコードを教えてください。
よろしくお願いします。

Answer 1

もっとシュッとしたやり方ありそうやの〜とは思いつつ、こんなんどないでっか？

def remove_duplicates(a, b):
  s, unique_a, unique_b = set(), [], []
  for x, y in zip(a, b):
    if not x in s:
      s.add(x)
      unique_a.append(x)
      unique_b.append(y)
      
  return unique_a, unique_b

ってゆうのを作っておいて、

a = [2, 1, 2, 3]
b = ['a', 'b', 'c', 'd']

a, b = remove_duplicates(a, b)

とする　➡ サンプル

Comments

[man_]

御二人ともありがとうございます。
無事実装することができました。

[退会済みユーザー]

関数の中の

unique_b

という変数名はアカンかった。ちょっと長いけど、ちゃんと書くと

b_for_ unique_a

とかそないな感じにしたほうがええね。

Answer 2

あらかじめ削除すべき要素のリスト上での位置を取得しておき
そのあとに実際に削除（必要な要素のみ残す）すればよいです。

a=[2,1,2,3]
b=list('abcd')
N = len(a)

for i in range(N):
    # a[i] が重複していたら:
        # iを「削除すべき要素位置のリスト」に追加

a = [a[i] for i in range(N) if i は「削除すべき要素位置のリスト」に含まれない ]
b = [b[i] for i in range(N) if i ～]
print(a)# [2, 1, 3]
print(b)# ['a', 'b', 'd']

なお、重複削除をpandasに任せる手もあります。

import pandas as pd

a=[2,1,2,3]
b=list('abcd')
c=list('wxyz')

sr = pd.Series(a).drop_duplicates()
a, b, c = [pd.Series(l).iloc[sr.index].tolist() for l in [a,b,c]]
print(a)# [2, 1, 3]
print(b)# ['a', 'b', 'd']
print(c)# ['w', 'x', 'z']