python - 将列添加到包含其他列值列表的 pandas DataFrame

Question

我有一个 DataFrame，我需要向其添加一列。该列必须是包含两个值的列表:

当前表:

    lat  long  other_value
0   50   50    x
1   60   50    y
2   70   50    z
3   80   50    a

需要的表格:

    lat  long  other_value  new_column
0   50   50    x            [50, 50]
1   60   50    y            [60, 50]
2   70   50    z            [70, 50]
3   80   50    a            [80, 50]

我知道这非常简单，但文档似乎没有涵盖这一点(至少显然没有)。

Answer 1

一种方法是使用tolist():

>>> df['new_column'] = df[['lat', 'long']].values.tolist()
>>> df
   lat  long other_value new_column
0   50    50           x   [50, 50]
1   60    50           y   [60, 50]
2   70    50           z   [70, 50]
3   80    50           a   [80, 50]

但总的来说，我会非常谨慎地在 DataFrame 中使用列表，因为它们在列中更难操作，而且您无法获得整数/ float 带来的许多性能优势。