如何将numpy.ndarray中的nan值转换为字符串类型？

你的X_train是混合了字符串和float类型NaN的numpy object数组，要把这些float类型的NaN转换为字符串，有两种简洁的实现方式：

方法一：自定义转换函数 + np.vectorize #

通过自定义函数判断元素类型和是否为NaN，再用np.vectorize批量处理数组：

import numpy as np

def convert_nan_to_str(x):
    if isinstance(x, float) and np.isnan(x):
        return 'NaN'  # 替换成你需要的任意字符串，比如示例中的长文本
    return x

# 处理数组
X_train_processed = np.vectorize(convert_nan_to_str)(X_train)

方法二：掩码筛选 + np.where #

先创建掩码定位所有float类型的NaN，再用np.where批量替换：

import numpy as np

# 生成掩码：标记所有float类型的NaN元素
nan_mask = np.vectorize(lambda x: isinstance(x, float) and np.isnan(x))(X_train)
# 替换掩码匹配的元素为目标字符串
X_train_processed = np.where(nan_mask, 'NaN', X_train)

验证结果 #

处理后执行你原来的测试代码：

for i in X_train_processed:
    if isinstance(i, float):
        print(i)

此时不会再输出nan，所有原float类型的NaN已转换为指定字符串。

内容的提问来源于stack exchange，提问作者Vineet Vinayak