شرط داخل DataFrame

وعليكم السلام ورحمة الله،

المطلوب هو جمع عمودين في DataFrame ولكن مع شرط معين، بمعنى آخر نريد اختبار شرط معين على كل صف والحصول على قيمة بناء على نتيجة هذا الشرط، لذلك نقوم أولًا بإنشاء الوظيفة التي سنطبقها على كل صف، ثم نستخدم الوظيفة apply ونعطيها الوظيفة التي أنشأناها

import pandas as pd

df = pd.DataFrame({'A': [10, 20, 50, 60, None, 80], 'B': [5, 10, 15, 20, 25, 30]})

# إنشاء وظيفة الاختبار المطلوب تطبيقه على كل صف
# يجب ملاحظة أن هذه الوظيفة سيتم استدعاؤها مع كل صف
def sum_with_condition(row):
    # نختبر إذا كانت قيمة الخانة الأولى تحتوي على بيانات وأيضًا قيمتها أكبر من 50
    if pd.notnull(row['A']) and row['A'] >= 50:
        # إذا تحقق الشرط نقوم بجمع الخانة بالعمود الأول مع الخانة بالعمود الثاني
        return row['A'] + row['B']
    else:        
        return 0
      
# إنشاء عمود جديد وتكون قيمته هو الناتج العائد من الوظيفة
# sum_with_condition
# التي يتم تطبيقها على كل صف 
df['C'] = df.apply(sum_with_condition, axis=1)

print(df)