python - pandas - 如何使用单元格中的数组

Question

我有一个带有数组的数据框。示例:

df = pd.DataFrame([('a', [1,2,3]), ('b', [4,5,6])], columns=['name', 'values'])

  name     values
0    a  [1, 2, 3]
1    b  [4, 5, 6]

我知道 values 列数组的长度相同。我想计算值数组的 axis=0 上的平均值。在 numpy 中我可以这样做:

np.array([[1,2,3], [4,5,6]]).mean(axis=0) # result: array([2.5, 3.5, 4.5])

普通 Pandas 有可能吗？
如果没有，我如何轻松地将值 col 转换为 numpy 数组？我试过 df['values'].values，但这没有给出矩阵:

array([list([1, 2, 3]), list([4, 5, 6])], dtype=object)

Answer 1

使用 pandas series.tolist 将 pandas 系列转换为列表

np.array(df['values'].tolist()).mean(axis = 0)

#o/p
array([2.5, 3.5, 4.5])