擅长:python、mysql、java
<p>您可以使用<code>groupby</code>和<code>cumsum</code>列上的<code>cumsum</code>以及<code>shift</code>列上的<code>shift</code>来执行此操作,该列是刚刚创建的:</p>
<pre><code>df['AI_AFTER'] = df['AVAILABLE_INVENTORY'] - df.groupby('PART')['DEMAND'].cumsum()
df['AI'] = df.groupby('PART')['AI_AFTER'].shift().fillna(df['AVAILABLE_INVENTORY'])
print (df)
PART AVAILABLE_INVENTORY DEMAND AI_AFTER AI
1 A 12 6 6 12.0
2 A 12 2 4 6.0
3 A 12 1 3 4.0
4 B 24 1 23 24.0
5 B 24 1 22 23.0
6 B 24 4 18 22.0
7 B 24 3 15 18.0
</code></pre>