我有一本字典:
groupings = {'Group_A':{'species':['plantain', 'banana'], 'shape':'oblong'},
'Group_B':{'species':['lemon', 'orange'], 'shape':'round'}
}
我也有一个dataframe:
d = pd.dataframe(data={'orange': [1, 2], 'bannana': [3, 4], 'lemon': [3, 4]})
我想根据字典重命名数据框并求和:
#something like this:
d.rename(columns=groupings, inplace=True)
d = d.groupby(by=d.columns, axis=1).sum()
预期输出:
Group_A Group_B
3 4
4 6
我如何重塑字典以便它被pandas.DataFrame.rename()识别?
谢谢!
3条答案
按热度按时间ryevplcw1#
您可以使用字典理解来扁平化
species
的值6psbrbz92#
您需要创建dataframe中的列名到字典中指定的所需列名的Map:
8iwquhpp3#
试试这个:
这应该适用于多个组具有相同物种的情况。
输出: