Pythonの配列で重複する要素を変更したい

Question

lis = ["hoge","hogehoge","fuga","fugafuga","hoge","fuga"]

for i in lis:
  if i in lis[:lis.index(i)]:
    i += "_2"
print(lis)

のようなコードで

["hoge","hogehoge","fuga","fugafuga","hoge_2","fuga_2"]

のような出力を得たいですがうまくいかないです（エラーは出ないが出力が変わらない）。各要素の最大重複は２なので一回判定できればいいのですが...

Answer 1

ベストアンサー決定後に失礼します。
他の回答者の方々の回答を参考に、リスト内包表記で書いてみました。

[x + ['', '_2'][x in lis[0:i]] for i, x in enumerate(lis)]

• 動作確認用Repl.it: https://repl.it/@jun68ykt/Q231461

Answer 2

重複数が 2 より大きい場合や _2 に変更した文字がすでに元のリストに存在していた場合にも対応させてみました。

tt.py

def seq_for_duplicat(lis):
    ans = []
    for x in lis:
        if not x in ans:
            ans.append(x)
            continue

        idx = 2
        while True:
            x_seq = x + "_" + str(idx)
            if not x_seq in ans:
                break
            idx += 1
        ans.append(x_seq)

    return ans

lis = ["hoge", "hogehoge", "fuga","fugafuga", "hoge", "fuga"]
print(lis)
print(seq_for_duplicat(lis))

lis = ["hoge", "hogehoge", "fuga","fugafuga", "hoge", "fuga",
       "hoge", "fuga", "hoge", "fuga",
       "hoge_2", "hoge_2", "hoge_2_3"]
print(lis)
print(seq_for_duplicat(lis))

実行例

追記：
jun68ykt さんの回答を元に、一行にしてみました。

>>> lis = ["hoge","hogehoge","fuga","fugafuga","hoge","fuga", "hoge","fuga"]
>>> [x + "_" + str(lis[0:i].count(x) + 1) if lis[0:i].count(x) > 0 else x for i, x in enumerate(lis)]
['hoge', 'hogehoge', 'fuga', 'fugafuga', 'hoge_2', 'fuga_2', 'hoge_3', 'fuga_3']
>>> 

Answer 3

lis = ["hoge","hogehoge","fuga","fugafuga","hoge","fuga"]

for i, x in enumerate(lis):
    before = [y for y in lis[:i] if y == x]
    if len(before) > 0:
        print(x + "_" + str(len(before) + 1))
    else:
        print(x)

結果：

hoge
hogehoge
fuga
fugafuga
hoge_2
fuga_2

Answer 4

ループしながらin-placeで処理するのはPythonでは意外と面倒です。i += "_2"で変数iに束縛されている文字列は望みのものになりますが、iに新しい文字列オブジェクトが割り当てられるだけでリストには影響しません。どうしてもやりたければ、添字を使います。

また、そのロジックもうまく動いていないようです（indexの数字を見てみましょう）。

ということで以下の方針で書いてみました。

• in-placeで処理するのはやめて、別に結果のリストを作ることにする
• index使うとかえって難しいし線形探索で遅いのでsetを使う

>>> lis = ["hoge","hogehoge","fuga","fugafuga","hoge","fuga"]
>>> s = set()
>>> result = []
>>> for x in lis:
...     if x in s:
...         result.append(x + "_2")
...     else:
...         s.add(x)
...         result.append(x)
... 
>>> result
['hoge', 'hogehoge', 'fuga', 'fugafuga', 'hoge_2', 'fuga_2']

Answer 5

元のコードを活かすなら、こんな感じでしょうか。

lis = ["hoge","hogehoge","fuga","fugafuga","hoge","fuga"]

for i, e in enumerate(lis):
  if e in lis[:i]:
    lis[i] += "_2"

print(lis)

lisがもっと長いなら、次のように別途記憶領域を用意した方が効率的です。

lis = ["hoge","hogehoge","fuga","fugafuga","hoge","fuga"]

appeared = set()
for i, e in enumerate(lis):
  if e in appeared:
    lis[i] += "_2"
  else:
    appeared.add(e)

print(lis)

追記

こんなふうにも書けます。

work_set = set()
lis = [e + '_2' if e in work_set else work_set.add(e) or e for e in lis]
print(lis)

でもあんまり行儀の良いコードでは無いですね。避けた方が良いでしょう。