如何根据多索引数据帧中的行索引值创建列？

Question

假设您有以下数据帧：

ridx = pd.MultiIndex.from_tuples([('Tom','apple'),('Tom','banana'),('Anna','banana'),('Anna','citrus')], names=['basket', 'fruit']) 
df = pd.DataFrame({'Quantity' : [1, 2, 3, 4]}, index=ridx)
df

你有一本字典可以告诉你每个水果含多少糖：

sugar_content = {'apple':3,'banana':10,'citrus':2}

如何向dataframe添加一个新列，其中包含“水果”索引中指定的水果的糖含量？你知道吗

Answer 1

您可以使用df.index.get_level_values()检索索引值数组，然后在字典的get方法上使用np.vectorize()来执行映射：

fruits  = df.index.get_level_values('fruit').values
fruits_sugar = np.vectorize(sugar_content.get)(fruits) # vectorize the get method of the dictionary and pass the sugar_content ndarray
df['sugar per fruit'] = fruits_sugar
df