python - 通过遍历索引列 : python 中的重复项来运行 for 循环

Question

我是Python的业余用户，想要开发一种逻辑来找出可以从与零件相关的可用数量中满足的订单。

partno  ordernumber orderqty    Qty available
A        abc          15        20
A        acb           5        20
A        adc           8        20
B        a41           9        60
B        a14          16        60
B        a56          30        60

各个零件号都有唯一的订单号和订单数量。可用数量是与零件编号对应的可用总数量。

我想从相关零件的可用数量中找出可以满足的订单。输出将类似于以下内容

partno  ordernumber orderqty    Qty available  Y/N
A        abc          15        20              Y
A        acb           5        20              Y
A        adc           8        20              N
B        a41           9        60              Y
B        a14          16        60              Y
B        a56          30        60              Y

我正在努力编写一个 for 循环，它将遍历重复的零件号，并在每次迭代订单号时从可用数量中减少 orderqty。

伪代码:

      for partno in df
      for ordernumber in df.interrow
      if (qtyavailble>orderqty) 
      {df.newcoln = "Y" & qtyavailable = qtyavailable-orderqty}
      else {df.newcoln="N"}

我知道这是一个可怕的代码，但你明白了。谢谢!

Answer 1

可以groupbypartno，然后对orderqty进行cumsum，看看它是否大于Qtyavailable

df.groupby('partno').orderqty.cumsum().le(df.Qtyavailable).map({True:'Y',False:'N'})
Out[459]: 
0    Y
1    Y
2    N
3    Y
4    Y
5    Y
dtype: object