使用numpyの - 基本的
a = df.values
out = a[np.abs(a[:,1:] - dfB.values.ravel()).argmin(0),0]
、我々はdfB
を引きますdfA
の各行から、私たちはNumPy配列を使って作業しているので(私たちはそれらをwi th .values
)、これはbroadcasted manner
で効率的に減算されます。次に、絶対値を見つけ、.argmin(axis=0)
、つまり.argmin(0)
で各列に沿ってarg-minimumを探します。
NaNs
で作業している場合は、これらを無視するとnp.nanargmin
を使用してください。この、jezraelため
# Extract array from dfA
In [9]: a = dfA.values
# Slice a from col-1 onwards and perform broadcasted differencing with dfB values
In [10]: a[:,1:] - dfB.values.ravel()
Out[10]:
array([[-0.92 , -0.802, -0.642, -0.523],
[-0.02 , -0.042, -0.022, -0.013],
[-0.02 , -0.042, -0.012, -0.003],
[ 0. , -0.022, 0.008, 0.017],
[ 0.02 , -0.002, 0.028, 0.037],
[ 0.04 , 0.018, 0.048, 0.057]])
# Get absolute values
In [11]: np.abs(a[:,1:] - dfB.values.ravel())
Out[11]:
array([[ 0.92 , 0.802, 0.642, 0.523],
[ 0.02 , 0.042, 0.022, 0.013],
[ 0.02 , 0.042, 0.012, 0.003],
[ 0. , 0.022, 0.008, 0.017],
[ 0.02 , 0.002, 0.028, 0.037],
[ 0.04 , 0.018, 0.048, 0.057]])
# Look for argmin along each col
In [14]: idx = np.abs(a[:,1:] - dfB.values.ravel()).argmin(axis=0)
In [17]: idx
Out[17]: array([3, 4, 3, 2])
# First col from a
In [15]: a[:,0]
Out[15]: array([ 0., 1., 2., 3., 4., 5.])
# Index into first col with those indices to select the desired output values
In [16]: a[idx,0]
Out[16]: array([ 3., 4., 3., 2.])
ええ、主にタイミングの違いは 'NaN'の扱いだから...どう思いますか? – jezrael
@jezrael私はそれが配列のほうが速いと思うだろうと思いますか? Nanの処理は 'np.nanargmin()'で解決できます。 – Divakar
@jezrael知っているとpandasはこれらの減算に 'broadcast'を使いますか? – Divakar