python pandas data 座標 距離

Question

実現したいこと

データはセルの中心座標を表しており、以下のような計算をしたいです。
実際のデータは以下のように規則正しくないです。

元データ
df = pd.DataFrame

||x|y|z|
|:--|:--:|--:|
|0|1|1|1|
|1|3|3|3|
|2|5|5|5|
|3|7|7|7|
|4|9|9|9|
|||||
|n|x|y|z|

↓
|x0| = 2x0
|x1| = 2(x1-|x0|)　= 2*(x1-sum(|x0|:|x0|))
|x2| = 2*(x2-(|x0|+|x1|)) = 2*(x2-sum(|x0|:|x1|))
|x3| = 2*(x3-(|x0|+|x1|+|x2|)) = 2*(x3-sum(|x0|:|x2|))
|x4| = 2*(x4-(|x0|+|x1|+|x2|+|x3|)) = 2*(x4-sum(|x0|:|x3|))

|n| =

上記のような計算をx,y,zそれぞれ行いたいです。

求めたいデータ
↓
||x|y|z|
|:--|:--:|--:|
|0|2|2|2|
|1|2|2|2|
|2|2|2|2|
|3|2|2|2|
|4|2|2|2|
|~|~|~|~|
|n|x|y|z|

Answer 1

applyを使って素直に実装すると以下のような感じでできます。

import pandas as pd

df = pd.DataFrame({'X':list(range(1,10,2))})

v_sum = 0
def f(v):
    global v_sum
    vn = 2*(v-v_sum)
    v_sum += vn
    return vn

df['X2'] = df['X'].apply(f)
print(df)
"""
   X  X2
0  1   2
1  3   2
2  5   2
3  7   2
4  9   2
"""

Comments

[Itasca]

すごい。。。
ありがとうございます。