如何從數據框創建列表？

Question

我有一個坐標為x0，y0，x1，y1的數據框。 我正在嘗試獲取坐標列表

datafrme就像-

      x0     y0      x1      y1      x2      y2 ...
 1  179.0   77.0    186.0   93.0    165.0   91.0... 
 2  178.0   76.0    185.0   93.0    164.0   91.0...

期望的輸出

 [  [ [179.0,77.0], [186.0,93.0], [165.0,91.0  ],...  ],
    [ [178.0,76.0 ],[185.0,93.0 ],[164.0,91.0 ] ,...  ]
 ]

我正在嘗試按行獲取坐標列表

我已經試過了

df = pd.read_csv(path,header=None)
df.reset_index(drop=True, inplace=True)
data = df.values.tolist()

感謝您提供的所有見解！

Answer 1

[[[y['x0'], y['y0']], [y['x1'], y['y1']], [y['x2'], y['y2']]]for x,y in df.iterrows()]

df這將是您的數據框

輸出將類似於您描述的格式

 [  [ [179.0,77.0], [186.0,93.0], [165.0,91.0  ],...  ],
    [ [178.0,76.0 ],[185.0,93.0 ],[164.0,91.0 ] ,...  ]
 ]

Answer 2

如果性能很重要，請使用reshape ：

a = len(df.columns)
b = len(df)

L = df.values.reshape(b, a).reshape(b, a // 2, 2).tolist()
print (L)
[[[179.0, 77.0], [186.0, 93.0], [165.0, 91.0]], 
 [[178.0, 76.0], [185.0, 93.0], [164.0, 91.0]]]