Как проверить NaN в numpy ndarray, который содержит строки - PullRequest
0 голосов
/ 01 апреля 2020

Обычно я могу проверить NaN с помощью numpy .isnan ():

import numpy as np

arr1 = np.array([[1, 2], [3, 4], [np.nan, 5]])
print(type(arr1))
print(np.isnan(arr1))

<class 'numpy.ndarray'>
[[False False]
 [False False]
 [ True False]]

Но как я могу добиться того же для массива, который содержит строки?

arr2 = np.array([[1, 2], [3, 4], [np.nan, 5], ['high', 'low']])
print(type(arr2))
print(np.isnan(arr2))

<class 'numpy.ndarray'>

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-19-4a4a4bc72076> in <module>
      1 array = np.array([[1, 2], [3, 4], [np.nan, 5], ['high', 'low']])
      2 print(type(array))
----> 3 print(np.isnan(array))

TypeError: ufunc 'isnan' not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''

1 Ответ

0 голосов
/ 01 апреля 2020

Как предложили комментаторы, включение строк в массив привело к тому, что все элементы были принудительно принудительно приведены к строкам, включая значения np.nan, которые теперь равны 'nan', и которые можно найти с помощью arr2 == nan :

print(arr1=='nan')

[[False False]
 [False False]
 [ True False]
 [False False]]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...