Как я могу вычислить среднее значение, отбросив NaN и выбросы из фрейма данных в этом формате?

У меня есть фрейм данных в формате ниже:

Original Dataframe

    |  x  |  value1  |  value2  |  value3  |  value4
 ---|-----|----------|----------|----------|-----------
  0 |  1  |    1     |   NaN    |    3     |   1
  1 |  2  |    4     |   NaN    |    1     |   NaN
  2 |  3  |    2     |    6     |    1     |   2
  3 |  4  |    1     |    1     |    2     |   1

Моя цель - найти среднее значение для каждой строки, отбросив NaN, а также исключив выбросы. Цель состоит в том, чтобы создать новый фрейм данных со следующим форматом:

Desired Dataframe

    |  x  |  mean (after dropping the NaN and the outliers)*
 ---|-----|--------
  0 |  1  |   a
  1 |  2  |   b   
  2 |  3  |   c   
  3 |  4  |   d   

* Обратите внимание, что a, b, c, d - средние значения (я не рассчитывал результат)


Конечная цель после достижения результата - напечатать график значений с течением времени.


Работая с одним списком значений, например, эквивалентом одного столбца значений и одного столбца индексов, я могу выполнять все операции: отбрасывать NaN, вычислять Z-оценку и затем возвращать список значений, как показано ниже:

import pandas as pd   
import numpy as np
from scipy import stats

data = {'value': [1, 2, 15, np.NaN, 2, 2, 2, 3, 1, 1], 
        'x': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}
df = pd.DataFrame(data)

df.dropna(inplace=True)
df = df[(np.abs(stats.zscore(df['return'])) < 2)]

Есть несколько проблем, которые возникают, когда я пытаюсь работать массово (и я уверен, что они возникают, потому что я привык программировать с использованием циклов, и должен быть «умный» способ выполнить эту операцию в Pandas.)

Я не могу рассчитать оценку Z, когда есть значение NaN. У меня такая ошибка:

/home/jupyterlab/conda/lib/python3.6/site-packages/ipykernel_launcher.py:14: RuntimeWarning: invalid value encountered in less

Итак, я знаю, что мне нужно удалить NaN. Но я не могу сделать это в этом формате фрейма данных, поскольку это повлечет за собой удаление строки или столбца, что приведет к потере информации.

Другой подход, который я безуспешно пробовал, заключался в преобразовании этой таблицы в длинную форму, что означало, что у меня было бы:

    |  x  |  valueName  |  actualValue
 ---|-----|-------------|--------------
  0 |  1  |  value1     |      1       
  1 |  1  |  value2     |     NaN    
 ...  ...      ...            ...
  2 |  2  |  value2     |      4  
  3 |  2  |  value2     |     NaN

Это заставило меня отказаться от Nan, но использовать Z-оценку было сложнее. Я уверен, что это обычная проблема, но я не могу понять, как ее решить.


person Dan    schedule 27.03.2019    source источник
comment
как насчет изменения NaN на 0?   -  person Sociopath    schedule 27.03.2019
comment
Я думаю, что на ваш вопрос был дан ответ здесь   -  person Yohai Magan    schedule 27.03.2019
comment
Возможный дубликат как zscore нормализовать столбец pandas с помощью nans?   -  person Yohai Magan    schedule 27.03.2019
comment
Привет, Ахшай Неврекар, если я это сделаю, среднее значение будет под влиянием. В этом случае, 0,0 или NaN означает запрет на вход.   -  person Dan    schedule 27.03.2019
comment
Привет, yochay magan, ссылка, которую вы опубликовали, относится к одномерному ряду, с которым я мог бы справиться, просто используя метод dropna.   -  person Dan    schedule 27.03.2019


Ответы (1)


Вы можете использовать:

from scipy import stats

#reshape to MultiIndex Series for remove NaNs
s = df.set_index('x').stack()
print (s)
x        
1  value1    1.0
   value3    3.0
   value4    1.0
2  value1    4.0
   value3    1.0
3  value1    2.0
   value2    6.0
   value3    1.0
   value4    2.0
4  value1    1.0
   value2    1.0
   value3    2.0
   value4    1.0
dtype: float64

#count zsore by first level of group - by x
s1 = s.groupby(level=0).transform(lambda x: np.abs(stats.zscore(x)))
print (s1)
x        
1  value1    0.707107
   value3    1.414214
   value4    0.707107
2  value1    1.000000
   value3    1.000000
3  value1    0.390567
   value2    1.692456
   value3    0.911322
   value4    0.390567
4  value1    0.577350
   value2    0.577350
   value3    1.732051
   value4    0.577350

#filter by condition and get mean by first level x, convert to DataFrame
s2 = s[s1 < 2].mean(level=0).reset_index(name='mean')
print (s2)
   x      mean
0  1  1.666667
1  2  2.500000
2  3  2.750000
3  4  1.250000
person jezrael    schedule 27.03.2019