Pythonの内包表記(for if)について

Question

Python初心者です。内包表記についての質問です。

空でない数字のリスト中から、重複の無いもののみを除外する課題の解決を試みました。

入力[1, 2, 3, 1, 3]→出力[1, 3, 1, 3]
入力[1, 2, 3, 4, 5]→出力[]

これに対して、次のようなコードを書きました。

def solve(data):
    
    return data.remove(uni for uni in data if data.count(uni)==1)

引数dataのリストから、countで個数が1個だったもののみをremoveで取り除く、という処理を意図しています。しかし、[1, 2, 3, 1, 3]など数のリストを入力すると、

ValueError: list.remove(x): x not in list

とエラーが返ってきます。原因がわからないので、ご教授いただきたく思います。

なお、課題の解決自体は、次のコードで達成できました。

def solve(data):

    for i in data[:]:
        if data.count(i)==1:
            data.remove(i)
    
    return data

Answer 1

def solve(data):
    return [k for k in data if data.count(k)>1]
data = solve(data)
print(data)

あまりよくない書き方ですが、もとのコードに忠実に行けば、

def solve(data):
    return [data.remove(k) for k in [uni for uni in data if data.count(uni)==1]]
solve(data)
print(data)

Comments

[Hashmore]

ありがとうございます。参考になります。
回答後半、「もとのコードに忠実な書き方」を実行してみると、リスト中に一つしかない数の個数のNoneからなるリストが返ってきます。kがdata中に一つしかない数を全て抽出できているようなので不思議です。なんとなく、dataから一部削除して残されたものではなく、dataから取り除いたものそのものを返そうとしているように思えます。

Answer 2

原因についてはyhmrさんのとおり、対応方法についてはmkgreiさんのとおりでよいと思います。

ちょっと気になったので関数の引数内で内包表記した場合の型を確認してみました。
generatorのようですね。ジェネレータ式というのかな？
参考：pythonの内包表記を少し詳しく

def hoge(o):
    print(type(o)) # <class 'generator'>

data = [1,2,3,1,3]
hoge(uni for uni in data if data.count(uni)==1)

Comments

[mkgrei]

そうなのです。
なので、
data.remove(next(uni for uni in data if data.count(uni)==1]))
をすると1個だけ削除できて、エラーが出るまで反復するとお望みの結果が得られます。
python3だとlistで囲むとgeneratorが実行されるので、謎のトリックを使っています。

削除しながらfor文を回すとだいたい大変なことになりますが、うまく動いているコードはちゃんとdata[:]を回していて、初心者とは思えない高度な実装になっています。

[can110]

コメントありがとうございます。
質問後半、data[:]で別オブジェクトできてるのでなので○ですね。
見た目ヒヤヒヤなコードですが(w

[mkgrei]

やらかしていた過去の嫌な思い出が蘇ります。
pythonは書き方が自由すぎて、細心の注意を払わないといけないのがたまにキズです。
よくわからないテクニックばかり身につきますね。

Answer 3

list.remove()はイテラブルを引数に取らないからだと思います。

これがsum()のようなイテラブルを引数に取る関数だったら上手く動作しますが、list.remove()の中にジェネレータ式を書いてもダメですね。

ちなみにこれをジェネレータ式ではなく

data.remove( [ uni for uni in data if data.count(uni)==1 ] )

のように配列内包で書いても同様に上手く動作しません。なぜならlist.remove()に渡される引数はリストであり、dataリストは要素にリストを持たないからです。

Answer 4

[1, 2, 3, 4, 5]を入力として該当のsolve関数を実行した場合、リストからintの2ではなく、listの[2]を削除しようとしていることがエラーの原因ではないでしょうか。

Comments

[Hashmore]

ありがとうございます。
データ型等について色々勉強してきましたが、solveの引数がジェネレータ式であるところまでは理解できました。ここでこのジェネレータ式の引数（リスト内包表記ではない）によって、intの2ではなくlistの[2]を削除する処理になっていることについて、よろしければもう少し詳しくお聞きしたいですが、よろしいでしょうか。

[yhmr]

私の回答の、
> listの[2]を削除しようとしている
という点が間違いです。

正しくは、他の皆様の回答やHashmoreさんのコメントにもあります通り、ジェネレータ式ですね。リスト内包表記と勘違いしておりました…
混乱させてしまい、大変申し訳ございません。

間違えた回答となっておりますため、低評価の程よろしくお願いいたします。