【python】配列の要素の値を簡単に変える方法

はじめに

pythonで特定の要素の値を変える方法を簡単に実装しました。

問題

1 行目に整数 A, B, N が与えられます。
2 行目に N 個の整数 a_1, a_2, ..., a_N が与えられます。
N 個の整数のうち、その数が A だった場合、B に書き換えてください。
書き換えた N 個の整数を改行区切りで出力してください。

実装したソースコード

実装したソースコードは下記になります。

b,t,n = map(int,input().split())
array1 = list(map(int,input().split()))
for i in range(n):
    if array1[i] == b:
        array1[i] = t
    print(array1[i])

実行した結果

最後に

map関数を使えば簡易に実装できますね。