指定した数字がタプルの要素のどちらかに入っていればその数字以外をセットに追加していく

Question

def xs(a, b):
    zs = []
    for i in a:
        if b in i:
            zs.append(i)
    return zs
a = xs([(1, 3), (1, 5), (1, 6), (5,9)], 1)
print (a)
コード
```### 前提・実現したいこと

例えばxs([(1,3), (1,5), (1,6),(5.9)], 1)という値があるとき戻り値を
{3, 5, 6}としたいです。

これは学校の課題です。どうしても上のようにできなく色々調べながらやっていたところこのサイトを見つけたので思い切って質問してみました。
まだ習い始めたばかりですごいレベルの低い質問かもしれませんが回答していただけると嬉しいです。

課題をそのまま出すのはどうかと思ったので関数のところを少し変えています。

### 発生している問題・エラーメッセージ

エラーメッセージ

### 該当のソースコード

```ここに言語名を入力
ソースコード

試したこと

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[meg_]

コードは「コードの挿入｣で記入してください。

[Salasse]

すいませんでした。
修正しました。

[meg_]

課題とのことですが回答に制約等はありますか？ どんな方法でもOKなんですか？

[Salasse]

制約は始まりが　def xs(a, b):　から始まるのみです。
よろしくお願いします。

Answer 1

えーと、戻り値はセットですね。{3, 5, 6}と書いてあるし。
それから、「タプルの要素のどちらかに入っていれば」とあるので、タプルの先頭とだけ比べてはいけませんよね。

それらを踏まえると、質問者のコードを活かすのであれば、こんなんですかね。

def xs(a, b):
    zs = set()
    for i in a:
        if b in i:
            zs.add(i[1 if i[0] == b else 0])
    return zs

自分で書くなら、こうかな。

def hoge(a, b):
    return {i[1 if i[0] == b else 0] for i in a if b in i}

Comments

[Salasse]

回答ありがとうございます！！
まとまったコードも一緒に送っていただきとても勉強になりました。

Answer 2

宿題の丸投げはよろしくありませんが、ご自身なりに取り組まれているようですのでご協力を

各配列(1, 3)等の0個目がbかどうか、bであれば1個目の3をzsに格納するので

def xs(a, b):
  zs = []
  for i in a:
    if b == i[0]:
      zs.append(i[1])
  return zs

a = xs([(1, 3), (1, 5), (1, 6), (5,9)], 1)
print (a)

な感じになるかと思います。

Comments

[Salasse]

回答ありがとうございます！！
しっかり参考にさせていただきます。