２つのリストの中に一致する要素がある時、その値を"NaN"置換したい

Question

aのリストの中にある要素がbのリストにもあった場合、bのリストのその要素を"NaN"に書き換えたcというリストを作りたいです。
やりたいこと
a = [1, 3, 5, 8, 2, 9]
b = [2, 4, 6, 3, 7, 10]
を比較して以下のcリストを作る
c = [NaN, 4, 6, NaN, 7, 10]

for文を使って書いてみましたが、期待とは全く違う動きをしてしまいます。
何かいい方法があれば教えてください。どうぞ宜しくお願いします。

a = [1, 3, 5, 8, 2, 9]
b =[2, 4, 6, 3, 7, 10]


str_a = [str(x) for x in a]
str_b = [str(y) for y in b]

c = []
for x in str_a:
    for y in str_b:
        if x == y:
            rep = y.replace(y, "NaN")
            c.append(rep)
        else:
            c.append(y)
            
print(c)

Answer 1

a = [1, 3, 5, 8, 2, 9]
b = [2, 4, 6, 3, 7, 10]
c = [_ if _ not in a else "NaN" for _ in b]
print(c)
# ['NaN', 4, 6, 'NaN', 7, 10]

NaNは文字列の"NaN"でいいですかね。np.nanでは？

Comments

[Yumame]

出来ました！！！！！ありがとうございます！しかも本当にやりたかったのはnp.nanでした。察して頂いて感謝です。

Answer 2

nan = float('nan')

set_a = set(a)
c = [
    nan if e in set_a else e
    for e in b
]
print(c)

お二方の良いとこどりみたいな回答ですけれど、

• nanは組み込みで用意されている
• setはsetのまま使って初めて帰属演算に有利になる

ということは主張しておきたいと思います。

Comments

[Yumame]

ありがとうございます！仕組みがよく分かっていないことが多いので大変勉強になります。

Answer 3

fot 文を使った例では、以下で実現可能かと思います。

import numpy as np
import copy
a = [1, 3, 5, 8, 2, 9,11,2,6]
b = [2, 4, 6, 3, 7, 10]
c = copy.copy(b)
for i in range(len(c)):
    for j in range(len(a)):
        if c[i] == a[j]:
            c[i] = np.nan
print(c)

for 文を使わない場合は下記の方法でも実現可能かと思います。

import numpy as np
a = [1, 3, 5, 8, 2, 9]
b = [2, 4, 6, 3, 7, 10]
l_dup = list(set(a)&set(b)) #重複要素のリスト作成
def dup(i):
    if i in l_dup:
        i = np.nan
    return i
c = list(map(lambda i: dup(i), b))
print(c)

文字列の"NaN" は np.nan にして記載おります。
ご参考まで。

Comments

[Yumame]

回答ありがとうございました！for文の方はrange(len())を使ってindexでlistの要素を指定できるようにすれば良かったんですね。目から鱗です。自力では絶対思いつきませんでした。大変勉強になりました。