python pandas 遍历dataframe放到list中

img

我有一个dataframe是这样的,根据idx我想分成三组,每组都是np的array格式,最后将3个array放到一个list,请问该怎么做?谢谢
预期的最终格式就是这样:
[
array([
[5,3],
[5,3],
[5,3]
]),
array([
[6,3],
[5,3],
[5,3]
]),
array([
[5,3],
[5,3],
[5,3]
])
]

有点没明白,是不是以下这样子:

>>> import pandas as pd
>>> data = {'idx':[0,1,2,0,1,2,0,1,2],'lat':[5,5,5,6,5,5,5,5,5],'lon':[3,3,3,3,3,3,3,3,3]}
>>> df = pd.DataFrame(data)
>>> df
   idx  lat  lon
0    0    5    3
1    1    5    3
2    2    5    3
3    0    6    3
4    1    5    3
5    2    5    3
6    0    5    3
7    1    5    3
8    2    5    3
>>> df3 = [df[:3][['lat','lon']],df[3:6][['lat','lon']],df[6:][['lat','lon']]]
>>> df3[0]
   lat  lon
0    5    3
1    5    3
2    5    3
>>> df3[1]
   lat  lon
3    6    3
4    5    3
5    5    3
>>> df3[2]
   lat  lon
6    5    3
7    5    3
8    5    3