python – 从Pandas数据帧创建嵌套列表

2019年7月28日 189次阅读

我有一个简单的pandas数据框,有两列.我想生成这两列的嵌套列表.

geo = pd.DataFrame({'lat': [40.672304, 40.777169, 40.712196], 
                'lon': [-73.935385, -73.988911, -73.957649]})

我对此问题的解决方案如下：

X = [[i] for i in geo['lat'].tolist()]
Y = [i for i in geo['lon'].tolist()]

for key, value in enumerate(X):
    X[key].append(Y[key])

但是,我觉得必须有一个比这更好的方法.

谢谢！

最佳答案 大熊猫建立在numpy之上. DataFrame将其值存储在numpy数组中,该数组具有tolist方法.

>>> geo = pd.DataFrame({'lat': [40.672304, 40.777169, 40.712196],
    ...:                 'lon': [-73.935385, -73.988911, -73.957649]})
    ...:
>>> geo.values
>>>
array([[ 40.672304, -73.935385],
       [ 40.777169, -73.988911],
       [ 40.712196, -73.957649]])
>>>
>>> geo.values.tolist()
[[40.672304, -73.935385], [40.777169, -73.988911], [40.712196, -73.957649]]
点赞