Python Pandas Funktion mit zwei Argumenten rollen

Question

Liebe Mein Anfänger für Python ist eine harte Prüfung unterziehen ...

Ich brauche eine Funktion in einem Rollfenster einer festen Länge berechnen (sagen wir mal: 5). Die Funktion benötigt zwei Parameter. Ich bin mir der Antwort sehr bewusst, here die fast identisch ist, aber ich bekomme immer Fehler.

Mein Code ist einfach:

import numpy as np 
import pandas as pd 
import scipy as sp 
import scipy.stats 

df = pd.DataFrame({'A' : np.arange(20), 'B' : np.random.randint(0,20,20)}) 

def my_tau2(idx): 
    x = df.loc[idx, 'A'].astype('float') 
    y = df.loc[idx, 'B'].astype('float') 
    return scipy.stats.mstats.kendalltau(x, y)[0] ## breaks without this [0] 

pd.rolling_apply(np.arange(len(df), dtype = np.dtype('int16')), 5, my_tau2) 

Und ich bekomme immer folgende Fehlermeldung:

enter code 
File "<ipython-input-6-d6cbc608d2f0>", line 7, in <module> 
pd.rolling_apply(np.arange(len(df), dtype = np.dtype('int16')), 5, my_tau2) 
File "D:\Users\502031217\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\stats\moments.py", line 584, in rolling_apply 
kwargs=kwargs) 
File "D:\Users\502031217\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\stats\moments.py", line 240, in ensure_compat 
result = getattr(r, name)(*args, **kwds) 
File "D:\Users\502031217\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\core\window.py", line 863, in apply 
return super(Rolling, self).apply(func, args=args, kwargs=kwargs) 
File "D:\Users\502031217\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\core\window.py", line 621, in apply 
center=False) 
File "D:\Users\502031217\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\core\window.py", line 560, in _apply 
result = calc(values) 
File "D:\Users\502031217\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\core\window.py", line 555, in calc 
return func(x, window, min_periods=self.min_periods) 
File "D:\Users\502031217\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\core\window.py", line 618, in f 
kwargs) 
File "pandas\algos.pyx", line 1831, in pandas.algos.roll_generic (pandas\algos.c:51581) 
TypeError: a float is required 

Ich habe mit, dass zu kämpfen und ich werde verrückt. Meine Modulversionen sind:

• numpy: 1.11.0
• scipy: 0.17.1
• Pandas: 0.18.1
• Python: 3.5.1

Irgendwelche Hinweise w.r.t. Wie man das auf eine andere Weise korrigiert oder berechnet, ist von ganzem Herzen willkommen.

Answer 1

Ich bin nicht vertraut mit Kendall Tau Coefficient, aber nach dem oben verlinkten Post, vielleicht sollten Sie Ihre Tau-Funktion umschreiben, um nur einen Wert zurückzugeben. Also, durch den Link zu urteilen Sie zur Verfügung gestellt, würde ich Ihre tau entwirft wie folgt (noch nicht zu flexibel, meiner Meinung nach, da sie fest codierte Spaltennamen aus äußeren Umfang verwendet):

def my_tau2(idx): 
    df_tau = df[["A","B"]].iloc[idx] 
    return scipy.stats.mstats.kendalltau(df_tau["A"], df_tau["B"])[0] 

die mich erlauben auszuführen rolling_apply (und natürlich auch in die Datenrahmen sparend - die Sie nicht getan zu haben schien):

df["tau"] = pd.rolling_apply(np.arange(len(df)), 5, my_tau2) 

Laufen diese outputed folgendes Ergebnis:

 A B  tau 
0 0 0  NaN 
1 1 11  NaN 
2 2 2  NaN 
3 3 11  NaN 
4 4 17 0.737865 
5 5 9 0.105409 
6 6 5 0.000000 
7 7 9 -0.527046 
8 8 15 -0.105409 
9 9 11 0.527046 
10 10 4 0.000000 
11 11 6 -0.400000 
12 12 14 -0.200000 
13 13 19 0.600000 
14 14 0 0.200000 
15 15 19 0.316228 
16 16 9 -0.105409 
17 17 1 -0.316228 
18 18 13 0.200000 
19 19 16 0.000000