日本語の文字を含む配列をNaNに置き換える方法

あるnumpy配列(下のコードのarr)は各要素がstr型で値が入っており，さらに'欠損'という文字列も含んでいます．

やりたいこととしては，arrの'欠損'をNaNに変更し，全要素を実数型に変換したいのです．

素直にarr.astype(np.float32)とやると'ValueError: could not convert string to float: '欠損''と表示されてうまくいきません．また私の環境によりエディタの中で日本語が使えないため，np.where(arr == '欠損')などという処理ができず，欠損の扱いに困っています．どなたか解決法を教えていただけませんか？よろしくお願いします．

↓コード

python
1import numpy as np
2
3arr = np.array(['2.0','14.0','12.11','欠損'])
4float_arr = arr.astype(np.float32)
5print(arr)
6

↓出力結果です

python
1----> 4 float_arr = arr.astype(np.float32)
2      5 print(arr)
3
4ValueError: could not convert string to float: '欠損'

行動規範の内容に同意します

回答1件

ベストアンサー

こんな感じでどうですか。

python
1import numpy as np
2
3def f(s):
4    try:
5        return np.float32(s)
6    except:
7        return np.nan
8
9f_v = np.vectorize(f)
10arr = f_v(np.array(['2.0','14.0','12.11','欠損']))
11print(arr.dtype)
12print(arr)
13""" =>
14float32
15[ 2.   14.   12.11   nan]
16"""
17