Python3: 要素数が1つのsetから要素を楽に取り出したい

Question

Python3.7での質問です。

あるsetに含まれる要素が１つだけであり、その要素を取り出したいとします。
例えば下のコードの場合、集合sの要素'spam'を変数varに代入するような状況です。
方法を2つ考えましたがこれが普通の方法なのかわかりません。
より簡明な・一般的な方法はあるでしょうか。

s = set(['spam'])

#変数varに集合s内の'spam'を代入したい
#方法1
for item in s:
    var = item

#方法2
var = list(s)[0]

Answer 1

var = list(s)[0]

は普通だと思います。

---

質問の上のコードは

for var in s:
    pass

で十分です。これもありだと思います。

---

1つだと確定しているならアンパック代入で

var, = s

でもいいです。
シンプルですが、読みにくいかも。

"使っても構わない"と思う程度には見かけます。

---

setが大きくてlistを生成するコストが馬鹿にならない時(かつ、1つだけ取り出したい、なんてことはまずないのですけれど)でも安心して使えるのは

var = next(iter(s))

でしょう。

(イテレーションの)最初の1つだけ取り出す、という意図とマッチするコードだと思います。

---

なんでもいいと思います。

Comments

[退会済みユーザー]

回答ありがとうございます。沢山紹介して頂けて助かります。
変数一つに対してアンパック代入できるのは盲点でした。
シンプルさではこれが一番でしょうか。

Answer 2

元のset自体に変更を加えても良い場合 Nippunさんの方法で良いかと思いますが、set自体を変更したくない場合は、質問の方法１ですかね。

ただ質問のコードで書かれている代入部は不要なので

for var in s:
    break

で良い気がします。

方法２は今回のように要素が１つの場合は特に問題ないかと思いますが、この方法では set の内容を一度 list にコピーしますので、要素が多くなると無駄になるのであまりお勧めしません。

あとは、イテレータを使って

var = next(iter(s))

でも良いですね。

Comments

[退会済みユーザー]

回答ありがとうございます。
for文のループ変数をループの外で使えることを知りませんでした。
iter(), next()も存在を初めて知りました。（知らないことだらけだ・・・）

Answer 3

var = s.pop()
ですね

Comments

[hayataka2049]

ただこれだと副作用で「取り出して」しまうので、状況次第で不適切な場合があります。

[Nippun]

取り出したいと書いてあるのでPopで構わないと思ってました。

[退会済みユーザー]

回答ありがとうございます。
確かに質問文の「取り出したい」だと受け取り方によりますね。失礼しました。

setの.pop()メソッドは把握してませんでした。list等と違い引数は指定できず、ランダムに値を削除して返り値とするのですね。