Замена пропусков на медиану

a = np.array([[np.nan,2,3,4,5], [6,7,np.nan,9,10], [11,12,np.nan,np.nan,15], [np.nan,np.nan,18,19,np.nan], [21,22,23,np.nan,np.nan]])

a = np.array([[3.5,2,3,4,5], [6,7,8,9,10], [11,12,12,12,15], [17.5,17.5,18,19,17.5], [21,22,23,22,22]])

import numpy as np
a = np.array([[np.nan,  2.,  3.,  4.,  5.],
              [ 6.,  7., np.nan,  9., 10.],
              [11., 12., np.nan, np.nan, 15.],
              [np.nan, np.nan, 18., 19., np.nan],
              [21., 22., 23., np.nan, np.nan]])
print('Исходный массив')
print(a)
b = np.nanmedian(a, axis=1) 
print('\nМедианы по строкам: ', b)
 
inds = np.where(np.isnan(a)) # Индексы NaN
a[inds] = np.take(b, inds[0]) # Замена по строкам, inds[1] - замена по колонкам
print('\nРезультат')
print(a)

import pandas as pd
df = pd.DataFrame({'col1': [np.nan, 6, 11, np.nan, 21], 
                   'col2': [2, 7, 12, np.nan, 22],
                   'col3': [3, np.nan, np.nan, 18, 23],
                   'col4': [4, 9, np.nan, 19, np.nan],
                   'col5': [5, 10, 15, np.nan, np.nan]})

import pandas as pd
import numpy as np
 
N = np.nan
df = pd.DataFrame([[N,  2,  3,  4,  5],
                   [6,  7,  N,  9, 10],
                   [11, 12, N,  N, 15],
                   [N,  N,  18, 19, N],
                   [21, 22, 23, N,  N]],
                 columns = list('ABCDE'))
 
print('Исходный массив')
print(df)
 
b = df.median(axis=1) 
print('\nМедианы по строкам: ', list(b))
 
print('\nРезультат')
print(df.T.fillna(b).T)

df = df.apply(lambda x: x.fillna(x.median()), axis=1)