Adding take function for DataFrameGroupBy