groupby 有多个条件

Question

df  fruit  year sale  important
0   apple  2010  
1   apple  2011
2   apple  2012
3   apple  2013
4   apple  2014 True    Yes
5   apple  2015 True
6   apple  2017 True
7   apple  2018 True
7   apple  2019 
8   apple  2020 True    Yes
9   banana 2010
...

如何生成“重要”列？

是的，如果每个水果： (1) 当年有销售， (2) 前一年没有销售，并且 (3) 该年与上一个“重要”年之间至少有 3 年的差距。

Answer 1

如果这适用于您的情况，请尝试。假设 df 按水果和年份排序。

for i in df['fruit'].unique():
    df1 = df[(df['sale'] == 'True') & (df['sale'].shift() != 'True') & (df['fruit'] == i)]
    df1 = df1[(df1['year'].diff() >=3) | (df1['year'].diff().fillna(0) == 0)]
    df.loc[df.index.isin(df1.index), 'important'] = 'Yes'

打印：

    fruit   year    sale    important
0   apple   2010    None    NaN
1   apple   2011    None    NaN
2   apple   2012    None    NaN
3   apple   2013    None    NaN
4   apple   2014    True    Yes
5   apple   2015    True    NaN
6   apple   2017    True    NaN
7   apple   2018    True    NaN
8   apple   2019    None    NaN
9   apple   2020    True    Yes
10  banana  2010    True    NaN

我测试了几个随机组合的水果和年份，并且正在尽我所能检查。