python - pd.groupby 按列+列表

Question

我知道如何在多个列上使用 .groupby，即:

df.groupby(['col5', 'col2']).size() 

我还知道如何使用与我正在使用的dataframe不同/独立的列表来使用.groupby，即:

df.groupby(aList).size()    

但是我如何使用 .groupby 同时使用单独的 list 和 dataframe 的一列(或某些列)我正在与.即，正确的写法是什么:

df.groupby(aList, aColumnName).size()    

？谢谢。

Answer 1

您可以使用 + 将列表连接在一起:

#for list with one column name
df.groupby(aList + [aColumnName]).size()  
#for join 2 lists
df.groupby(aList + collist2).size()  
#for join with columns
df.groupby(aList + df.columns[:2].tolist()).size()