创建函数以根据日期时间计算值

Question

我有这样的数据：

activity_date	company_name	new_company_status	calling	visit	quotation	po
03/10/2022	ABC	Yes	Yes	No	No	No
04/10/2022	ABC	No	No	No	Yes	Yes
05/10/2022	DEF	No	Yes	Yes	No	No
06/10/2022	XYZ	Yes	No	Yes	Yes	No
07/10/2022	DEF	No	No	No	Yes	Yes
08/10/2022	XYZ	No	Yes	No	No	Yes

1. 我想创建一个函数来检查每个相同的company_name，其中至少有一个new_company_status 为“是”，然后变成“是”calling，并计算它的总和，即使在不同的日期。
2. 我想创建一个函数来检查每个相同的company_name，即使在不同的日期，calling 也是“是”，po 也是“是”。

   这是伪代码我创建的：

   1.

   for every same company name:
       if 'new_company_status' = 'Yes':
           check 'activity_date'        # for new company status if it is a Yes
       if 'calling' = 'Yes':
           check 'activity_date'        # for calling if it is a Yes
       if calling_date >= new_company_date:
           new_company to call =+ 1
   

   for every same company name:
       if 'calling' = 'Yes':
           check 'activity_date'        # for calling if it is a Yes
       if 'visit' = 'No':
           if 'quotation' = 'No':
               if 'po' = 'Yes':
                   check 'activity_date'   # for po if it is a Yes
       if po_date >= calling_date:
           call to po += 1
   

   预期产出:

   1. 1
   2. 3

      如何将伪代码编码成Python？谁能帮我？提前谢谢你。

Answer 1

假设一个pandas数据帧，你可以使用布尔运算和groupby.any：

m1 = df['new_company_status'].eq('Yes')
m2 = df['calling'].eq('Yes')

(m1&m2).groupby(df['company_name']).any().sum()

输出：1