python - 从 Pandas 数据框创建嵌套列表

Question

我有一个包含两列的简单 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])

但是，我觉得一定有比这更好的方法。

谢谢!

Answer 1

pandas 构建于 numpy 之上。 DataFrame 将其 values 存储在 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]]