Lignes de dataframe times colonne d'une autre dataframe

Nov 15 2020

J'ai essayé d'obtenir la moyenne pondérée de chaque produit pour chaque personne. Donc pour Tom, il devrait avoir 20x1.0 + 19x2.0 + 10x3.0, j'espère aussi avoir les poids * produit par produit également.

data = {'Name':['Tom', 'nick', 'krish', 'jack'], '1stproducts':[20, 21, 19, 18], '2ndproduct': [19, 28, 10, 10], 
        '3rdproduct': [10, 18, 20, 30]} 
df = pd.DataFrame(data) 
weights = {"weights": [1.0, 2.0, 3.0]}
df2 = pd.DataFrame(weights) 

J'ai essayé pd.DataFrame.multiply(df, df2, axis = 1), mais j'ai obtenu NaN pour toutes les valeurs.

Réponses

1 ColeHoward Nov 15 2020 at 03:13

L'alignement de l'index df2résoudra votre problème. Et référençant la colonne de poids dans df2.

df2 = pd.DataFrame(weights, index=['1stproducts', '2ndproduct', '3rdproduct'])
In [26]: df[['1stproducts', '2ndproduct', '3rdproduct']] * df2.weights
Out[26]:
1stproducts  2ndproduct  3rdproduct
0         20.0        38.0        30.0
1         21.0        56.0        54.0
2         19.0        20.0        60.0
3         18.0        20.0        90.0

Question similaire ici: Comment calculer la somme pondérée de tous les éléments d'affilée dans les pandas?