0

x , y , 粒子

23、25、0

12、15、0

54, 45, 0

32, 11, 1

21, 43, 1

43, 11, 1

excel中有3列的产量。第一列 x，第二列 y，第三列 ID。如果粒子具有相同的标识号（ID），我想减去具有相同 ID 号的粒子的 x 和 y。例如;

对于 ID = 0 的列，我希望它：

对于 x 应该是
23-12 = 11
11-54 = -43

对于 y 应该是

25-15 = 10

10-45 = -35

这是我为它编写的代码。第一个对于相同的 ID 号可以正常工作，但在第二个循环中出现范围错误。可能是什么问题呢。我该如何解决。

from pyexcel_ods import get_data,save_data

# -*- coding: utf-8 -*-

data = get_data("deneme.ods")

s,extract=0,0

for i in range(len(data[u'Sheet1'])-2):
if data[u'Sheet1'][i][2]==data[u'Sheet1'][i+1][2]:
    s+=1
    continue
else:
    for j in range(s+1):
        extract -= data[u'Sheet1'][j][1]
        extract = abs(extract)
    data[u'Sheet1'][i].append(extract)
    extract=0
    s=0

save_data('deneme1.ods',数据)

4

1 回答 1

0

如果您可以将文件加载到数据框中，则可以这样做。

import pandas as pd
from operator import sub
from functools import reduce

d = [["x", "y", "particle"], [23, 25, 0],[12, 15, 0], [54, 45, 0], [32, 11, 1], [21, 43, 1], [43, 11, 1]]

df = pd.DataFrame(d[1:], columns=d[0])

my_sub = lambda e: reduce(sub, e)

df2 = df.pivot_table(index = "particle", values=["x", "y"], aggfunc=my_sub)
print(df2)
于 2019-10-09T12:16:34.250 回答