如何通过ID选择数据帧中的行信息

Question

我是python的新手。我有一个像这样的大数据框：

    ID  x   y
0   1   x1  y1
1   0   x2  y2
2   0   x3  y3
3   2   x4  y4
4   1   x5  y5
5   2   x6  y6

我想在ID 1和2之间采用（x; y）对，在这样的数据框中：

    coordinates
0   (x1,y1), (x2,y2), (x3,y3), (x4,y4)
1   (x5,y5), (x6,y6)

我已经尝试过使用double进行迭代，但是计算起来很长。我怎么能得到这个东西？

Answer 1

您可以为此使用熊猫和熊猫：

import pandas as pd

d = {'x': ["x1", "x2", "x3"], 'y': ["y1", "y2", "y3"]}
df = pd.DataFrame(data=d)
df

my_pairs = []

for item in df.iterrows():
  my_pairs.append(item[1])

以这种方式访问​​它：

my_pairs[0]           # for the pair
my_pairs[0]["x"]      # for the x-component