Pythalex
9/27/2019 - 11:21 AM

Calculer le total d'une ligne d'un dataframe - Pandas

# Deux instances de 0 avec une target de chaque
# Deux instances de 1 avec chacune une target à 0
# Deux instances de 2 avec chacune une target à 1
table = np.array([
    [0, 0, 1],
    [1, 1, 0],
    [2, 1, 1]
])
table_df = pd.DataFrame(data=table[:, 1:], index=table[:, 0], columns=["target_0", "target_1"])

# ajout du total
table_df["total"] = table_df.sum(axis=1)
table_df

"""
   target_0  target_1  total
0         0         1      1
1         1         0      1
2         1         1      2
"""