python pandas,DF.groupby()。agg(),agg()中的列引用
agg與相同aggregate。可調(diào)用的是一次傳遞一次的列(Series對(duì)象)DataFrame。
您可以idxmax用來(lái)收集具有最大計(jì)數(shù)的行的索引標(biāo)簽:
idx = df.groupby(’word’)[’count’].idxmax()print(idx)
產(chǎn)量
worda 2an 3the 1Name: count
然后用于loc在word和tag列中選擇那些行:
print(df.loc[idx, [’word’, ’tag’]])
產(chǎn)量
word tag2 a T3 an T1 the S
請(qǐng)注意,idxmax返回索引 標(biāo)簽。df.loc可用于按標(biāo)簽選擇行。但是,如果索引不是唯一的-即,如果存在帶有重復(fù)索引標(biāo)簽的行-df.loc則將選擇帶有標(biāo)簽的所有行idx。所以,要小心,df.index.is_unique是True如果你使用idxmax與df.loc
或者,您可以使用apply。apply的callable傳遞了一個(gè)sub-DataFrame,它使您可以訪問(wèn)所有列:
import pandas as pddf = pd.DataFrame({’word’:’a the a an the’.split(), ’tag’: list(’sstTT’), ’count’: [30, 20, 60, 5, 10]})print(df.groupby(’word’).apply(lambda subf: subf[’tag’][subf[’count’].idxmax()]))
產(chǎn)量
worda Tan Tthe S
使用idxmax和loc通常比快apply,尤其是對(duì)于大型DataFrame。使用IPython的%timeit:
N = 10000df = pd.DataFrame({’word’:’a the a an the’.split()*N, ’tag’: list(’sstTT’)*N, ’count’: [30, 20, 60, 5, 10]*N})def using_apply(df): return (df.groupby(’word’).apply(lambda subf: subf[’tag’][subf[’count’].idxmax()]))def using_idxmax_loc(df): idx = df.groupby(’word’)[’count’].idxmax() return df.loc[idx, [’word’, ’tag’]]In [22]: %timeit using_apply(df)100 loops, best of 3: 7.68 ms per loopIn [23]: %timeit using_idxmax_loc(df)100 loops, best of 3: 5.43 ms per loop
如果你想有一個(gè)字典映射字標(biāo)簽,那么你可以使用set_index 和to_dict這樣的:
In [36]: df2 = df.loc[idx, [’word’, ’tag’]].set_index(’word’)In [37]: df2Out[37]: tagword a Tan Tthe SIn [38]: df2.to_dict()[’tag’]Out[38]: {’a’: ’T’, ’an’: ’T’, ’the’: ’S’}解決方法
關(guān)于一個(gè)具體問(wèn)題,說(shuō)我有一個(gè)DataFrame DF
word tag count0 a S 301 the S 202 a T 603 an T 54 the T 10
我想 為每個(gè)“單詞” 找到 具有最多“計(jì)數(shù)”的“標(biāo)簽” 。因此,回報(bào)將類似于
word tag count1 the S 202 a T 603 an T 5
我不在乎計(jì)數(shù)列或訂單/索引是原始的還是混亂的。返回字典{ ‘the’:’S’ ,…}很好。
我希望我能做
DF.groupby([’word’]).agg(lambda x: x[’tag’][ x[’count’].argmax() ] )
但這不起作用。我無(wú)法訪問(wèn)列信息。
更抽象地講, agg( function ) 中的 函數(shù) 將其視為 __什么?
順便說(shuō)一句,.agg()與.aggregate()相同嗎?
非常感謝。
相關(guān)文章:
1. python pandas移動(dòng)窗口函數(shù)rolling的用法2. Python Pandas的簡(jiǎn)單使用教程3. Python Pandas常用函數(shù)方法總結(jié)4. python pandas模糊匹配 讀取Excel后 獲取指定指標(biāo)的操作5. python pandas利用fillna方法實(shí)現(xiàn)部分自動(dòng)填充功能6. 解決python pandas讀取excel中多個(gè)不同sheet表格存在的問(wèn)題7. 利用python Pandas實(shí)現(xiàn)批量拆分Excel與合并Excel8. Python Pandas pandas.read_sql函數(shù)實(shí)例用法9. 對(duì)python pandas中 inplace 參數(shù)的理解10. Python pandas軸旋轉(zhuǎn)stack和unstack的使用說(shuō)明

網(wǎng)公網(wǎng)安備