python - Pandas 聚合有序点坐标

Question

我有以下 pandas 系列的有序点坐标:

>>> import pandas as pd
>>> d=pd.Series([(1,1),(2,2),(3,4),(4,6)])
>>> d
0    (1, 1)
1    (2, 2)
2    (3, 4)
3    (4, 6)
dtype: object
>>> 

我想创建一系列由这些点形成的线段。在本例中，将有 3 条线段(点 0 和 1、1 和 2 以及 2 和 3 之间。

我可以通过将原始系列与其“移位”版本连接起来，删除第一个空值，然后将函数应用于两列来创建下面的内容。

>>> pd.concat([d.shift(),d],1).dropna().apply(lambda x:tuple([x[0], x[1]]),1)
1    ((1, 1), (2, 2))
2    ((2, 2), (3, 4))
3    ((3, 4), (4, 6))
dtype: object
>>> 

有更好的方法吗？

Answer 1

我喜欢使用 zip 来实现此目的

pd.Series(list(zip(d.values[:-1], d.values[1:])))

0    ((1, 1), (2, 2))
1    ((2, 2), (3, 4))
2    ((3, 4), (4, 6))
dtype: object