输入数据集
对于上面的数据集，编写python或sql命令，使其返回以下内容
对于每一组assignmentid，如果正确的值是1，则在有多行具有相同assignmentid的情况下使用值1追加一个新列，然后仅当至少有3个1或最后一个1时才将1放入新列。预期输出（第一列仅用于ref，表示有多个assignment id的行
我的逻辑是

> a = [0,1,1,1,0,1,0,1] counter = 0 for i in a: 
>     if (a[i] == a[i+1]):
>       counter=counter+1
>     else:
>       counter=0 
>     if (counter>=3):
>       break if (counter==3):   print("Mastered") else:   print("WheelSpinning")

但是我不知道如何在数据集中运行这个函数，或者如何将它作为上述数据的sql来编写。