将cols列表中的NaN值替换为其他cols列表的平均值

import numpy as np

def replace_nan_with_mean(cols):
    # 计算其他cols列表的平均值
    means = []
    for col in cols:
        col_mean = np.nanmean(col)
        means.append(col_mean)
    
    # 替换NaN值为平均值
    for i in range(len(cols)):
        col = cols[i]
        for j in range(len(col)):
            if np.isnan(col[j]):
                col[j] = means[i]
    
    return cols