この回答では、「zスコア」に基づくソリューションと「IQR」に基づくソリューションの2つの方法を提供したいと思います。
この回答で提供されるコードは、単一のdim numpy
配列と複数のnumpy
配列の両方で機能します。
最初にいくつかのモジュールをインポートしましょう。
import collections
import numpy as np
import scipy.stats as stat
from scipy.stats import iqr
zスコアに基づく方法
このメソッドは、数値が3つの標準偏差の範囲外かどうかをテストします。このルールに基づいて、値が外れ値の場合、メソッドはtrueを返し、そうでない場合はfalseを返します。
def sd_outlier(x, axis = None, bar = 3, side = 'both'):
assert side in ['gt', 'lt', 'both'], 'Side should be `gt`, `lt` or `both`.'
d_z = stat.zscore(x, axis = axis)
if side == 'gt':
return d_z > bar
elif side == 'lt':
return d_z < -bar
elif side == 'both':
return np.abs(d_z) > bar
IQRベースの方法
このメソッドは、値がより小さいq1 - 1.5 * iqr
か大きいかをテストしますq3 + 1.5 * iqr
。これは、SPSSのプロットメソッドと同様です。
def q1(x, axis = None):
return np.percentile(x, 25, axis = axis)
def q3(x, axis = None):
return np.percentile(x, 75, axis = axis)
def iqr_outlier(x, axis = None, bar = 1.5, side = 'both'):
assert side in ['gt', 'lt', 'both'], 'Side should be `gt`, `lt` or `both`.'
d_iqr = iqr(x, axis = axis)
d_q1 = q1(x, axis = axis)
d_q3 = q3(x, axis = axis)
iqr_distance = np.multiply(d_iqr, bar)
stat_shape = list(x.shape)
if isinstance(axis, collections.Iterable):
for single_axis in axis:
stat_shape[single_axis] = 1
else:
stat_shape[axis] = 1
if side in ['gt', 'both']:
upper_range = d_q3 + iqr_distance
upper_outlier = np.greater(x - upper_range.reshape(stat_shape), 0)
if side in ['lt', 'both']:
lower_range = d_q1 - iqr_distance
lower_outlier = np.less(x - lower_range.reshape(stat_shape), 0)
if side == 'gt':
return upper_outlier
if side == 'lt':
return lower_outlier
if side == 'both':
return np.logical_or(upper_outlier, lower_outlier)
最後に、外れ値を除外する場合は、numpy
セレクターを使用します。
ごきげんよう。