大熊猫在大型数据平台上的GroupBy

Question

我有一个包含150列的dataframe。在150列中，有15列类型为string，其余都是浮动列。

dataframe包含15个字符串列组合的相同记录，rest中的值可能不同。我想在这15个字符串列上做groupby，并对所有浮动列做一个和。

如何在Python中轻松地做到这一点，而无需指定groupby语句中每个列的名称，因为dataframe是巨大的。

Answer 1

提示一下你的工作。可以使用此代码筛选字符串类型的列。

string_columns = list()    
for col in df.columns:
    if df[col].dtypes == 'O':
        string_columns.append(col)

最后，string_columns将包含具有对象数据类型的列名，可以用于groupby函数。

Answer 2

使用列表理解1过滤类型为object类型的列，并按函数使用组中的cols

cols = [col for col in df.columns if df[col].dtype == np.object]
df.groupby(cols).sum()