Pandas calculer une nouvelle colonne à travers plusieurs dataframes

Question

Je suis nouveau pour les pandas et je cherche à refaire ce qui suit à partir de plusieurs feuilles Excel avec plusieurs données.

Voici la structure de haut niveau:

Master_df
Master_UID | Component_ID_1 | Quantité_1 | ... | Component_ID_8 | Quantity_8

Component_type_1_df
ID_composant | ... | poids

Component_type_2_df
ID_composant | ... | poids

Master_UID [ 'Component_ID_1'] contient de « ID_composant à la fois Component_type_1_df et Component_type_2_df

maintenant dans Master_df, je voudrais créer colonne Weight_1 colonnes en fonction poids de la 2 Component_type_X_df .

multiplié par Master_df ['Quantity_1'].

Merci beaucoup.

Answer 1

Master_df['Weight_1'] = Master_df['Component_ID'].map(Component_type_1_df.set_index('Component_ID')['weight']) * Master_df['Quantity'] 

Sans avoir un ensemble de données à jouer avec, je ne peux pas garantir que cela fonctionne comme prévu, mais la fonction de carte devrait vous aider à arriver là où vous voulez aller.

Si vous ne voulez pas ajouter à la fin de maître la colonne de poids calculée df vous pouvez effectuer les opérations suivantes:

Master_df.insert({desired_index_position},'Weight_1', Master_df['Component_ID'].map(Component_type_1_df.set_index('Component_ID')['weight']) * Master_df['Quantity') 

En fait, je récemment une question similaire à ce sujet. J'ai posté un ensemble de données pour déranger afin que vous puissiez apprendre la fonction.

PANDAS vlookup against series with common index using map

Answer 2

Depuis les deux Component_type_*_df DataFrames ne pas ont la même structure, concaténer seulement les parties de ces DataFrames que vous avez besoin, puis fusionner le résultat avec Master_df.

cols = ['Component_ID', 'weight'] 
Components_df = pd.concat([Component_type_1_df[cols], Component_type_2_df[cols]], axis=0) 
Master_df = pd.merge(Master_df, Components_df, 
        left_on='Component_ID_1', 
        right_on='Component_ID', how='left') 
Master_df['weight'] = Master_df['weight'] * Master_df['Quantity_1'] 
Master_df = Master_df.rename({'weight':'Weight_1'}) 

Depuis que nous avons restreint les Component_type_*_df DataFrames à seulement deux colonnes, et une colonne est utilisée comme la clé de fusion, la fusion ajoute qu'une seule colonne supplémentaire, weight-Master_df.

Puisque vous ne souhaitez pas que la colonne weight fusionnée dans Master_df, le code effectue le calcul ci-dessus dans Master_df['weight'] mais renomme puis cette colonne Weight_1.