應用 function Pandas DataFrame 無 Lambda 參數

Question

df = pd.DataFrame([["Test", "Test123"]] * 3, columns=['A', 'B'])


def shorterstring(string, count): 
    return string[0:-count]


df["A"].apply(lambda x: x[0:2])
Out[614]: 
0    Te
1    Te
2    Te

df["A"].apply(shorterstring(df["A"], 2))
TypeError: 'Series' object is not callable

我想在帶有字符串的列上應用自定義 function 。 我設法讓它與 lambda 表達式一起使用，但是當我想應用 function 時它不起作用。 誰能幫助我如何做到這一點？

Answer 1

使用functools.partial功能：

In [128]: from functools import partial                                                                     

In [129]: df["A"].apply(partial(shorterstring, count=2))                                                    
Out[129]: 
0    Te
1    Te
2    Te
Name: A, dtype: object