根據另一列條目計算兩行的差異

Question

我有這樣的數據框：

A列	B列
祖	2.5
ab	1個
菲	0
祖	2個
ab	0.5
菲	0
紫外線	1個
祖	2個
ab	1個

我需要兩個計算“zu”和“ab”之間的差異並將其放置在新列上

A列	B列	差異
祖	2.5	1.5
ab	1個	0
菲	0	0
祖	2個	1.5
ab	0.5	0
菲	0	0
紫外線	1個	0
祖	2個	1個
ab	1個	0

我嘗試了 diff() 和 shift() Funktion，但沒有用。 你有什么建議嗎？

Answer 1

示例代碼

data = {'Column A': {0: 'zu', 1: 'ab', 2: 'fi', 3: 'zu', 4: 'ab', 5: 'fi', 6: 'uv', 7: 'zu', 8: 'ab'},
        'Column B': {0: 2.5, 1: 1.0, 2: 0.0, 3: 2.0, 4: 0.5, 5: 0.0, 6: 1.0, 7: 2.0, 8: 1.0}}
df = pd.DataFrame(data)

---

代碼

s = df['Column B'].diff(-1).where(df['Column A'].eq('zu'), 0)

s

0    1.5
1    0.0
2    0.0
3    1.5
4    0.0
5    0.0
6    0.0
7    1.0
8    0.0

將s設為 diff 列

df.assign(diff=s)

output

 Column A   Column B    diff
0   zu      2.5         1.5
1   ab      1.0         0.0
2   fi      0.0         0.0
3   zu      2.0         1.5
4   ab      0.5         0.0
5   fi      0.0         0.0
6   uv      1.0         0.0
7   zu      2.0         1.0
8   ab      1.0         0.0