удалите 'nan' или уменьшите длину массива numpy, если массив содержит nan после преобразования в numpy из pandas

Question

Я хочу удалить nan из массива numpy. Допустим, мой NumPy массив содержит:

np_array = ["123","pqr","123",nan,"avb", nan] 

Ожидаемый результат:

["123","pqr","123","avb"] 

Если мы делаем это в панд с помощью pandas.dropna() удаляет всю строку, которую я не хочу делать. Я просто хочу удалить значение и уменьшить размер массива.

Есть ли какой-либо возможный способ?

Answer 1

isnan() должен сделать трюк. Работа минимального примера о том, как это сделать:

>>> import numpy as np 
>>> np_array = np.array([1,2,3,np.nan,4]) 
>>> np_array 
array([ 1., 2., 3., nan, 4.]) 
>>> np_array = np_array[~np.isnan(np_array)] 
>>> np_array 
array([ 1., 2., 3., 4.]) 

Answer 2

Попробуйте

np_clean = [x for x in np_array if str(x) != 'nan'] 

Это удалит нана из списка

Answer 3

Вы не можете использовать np.isnan потому что NaN s являются строками в вашем массив, но вы можете использовать булево индексирование по сравнению со строкой: "nan":

>>> import numpy as np 
>>> np_array = np.array(["123","pqr","123",np.nan,"avb", np.nan]) 
>>> np_array[np_array != 'nan'] 
array(['1234', 'pqr', '123', 'avb'], 
     dtype='<U4') 

Answer 4

Это работает для числовых массивов.

filter(lambda x: np.isfinite(x), np.array([1,2,3,np.nan])) 

>>>[1.0, 2.0, 3.0]