我有一个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