Как заменить элементы в списке или более на nan на позицию списка nan в python3?

Question

a = [1,2,3,np.nan,4,6,8,np.nan,5,3,7]
b = [np.nan,2,3,5,7,np.nan,6,2,6,9]

for i in np.arange(0,12,1):
    if a[i] == np.nan or b[i]==np.nan:
        a[i] = np.nan
        b[i] =np.nan

Я новичок ie в python, мне нужны окончательные результаты:

a = [np.nan, 2,3,np.nan, 4, 6, np.nan, np.nan,5,3,7]
b = [np.nan,2,3,np.nan,7,np.nan,np.nan,2,6,9]

Но это не сработало. Большое спасибо за любое предложение

Answer 1

Вы используете numpy, так почему бы не работать с массивами numpy?

a, b = map(np.array, [a, b])
l = min(a.size, b.size)
# Get a mask of NaN cells
m = np.isnan(a[:l]) | np.isnan(b[:l])
# Set to NaN based on the mask
a[:l][m] = np.nan
b[:l][m] = np.nan

a
# array([nan,  2.,  3., nan,  4., nan,  8., nan,  5.,  3.,  7.])

b
# array([nan,  2.,  3., nan,  7., nan,  6., nan,  6.,  9.])

_{Я знаю, что это отличается от «ожидаемого результата» OP, но на основе их объяснение Я думаю, что в их выводе есть ошибка.}

Answer 2

if len(a) > len(b):
    for i, element in enumerate(b):
        if np.isnan(element) or np.isnan(a[i]):
            a[i] = np.nan
            b[i] = np.nan
else:
    for i, element in enumerate(a):
        if np.isnan(element) or np.isnan(b[i]):
            a[i] = np.nan
            b[i] = np.nan

Answer 3

Вот решение:

for i in range(min(len(a), len(b))):
    if np.isnan(a[i]) or np.isnan(b[i]):
        a[i] = np.NaN
        b[i] = np.NaN
        
print(a)
# [nan, 2, 3, nan, 4, nan, 8, nan, 5, 3, 7]

print(b)
# [nan, 2, 3, nan, 7, nan, 6, nan, 6, 9]