python - Pandas/Numpy 从数组列中获取矩阵

Question

我有一个带有一列列表的 Pandas 数据框。

df:

    inputs
0   [1, 2, 3]
1   [4, 5, 6]
2   [7, 8, 9]
3   [10, 11, 12]

我需要矩阵

array([[ 1,  2,  3],
      [ 4,  5,  6],
      [ 7,  8,  9],
      [10, 11, 12]])

一个有效的方法来做到这一点？

注意:当我尝试时 df.inputs.as_matrix()输出是

array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]], dtype=object)

具有形状 (4,) ，不是 (4,3)如预期的。

Answer 1

您可以将列转换为列表，然后应用 numpy 数组，如果列中的所有列表都具有相同的长度，这将生成一个二维数组:

arr = np.array(df.inputs.tolist())

#array([[ 1,  2,  3],
#       [ 4,  5,  6],
#       [ 7,  8,  9],
#       [10, 11, 12]])

arr.shape
# (4, 3)

或者其他选项使用 .values首先访问 numpy 对象，然后将其转换为 @piRSquared 评论的列表，使用给出的示例稍微快一点:

%timeit df.inputs.values.tolist()
# 100000 loops, best of 3: 5.52 µs per loop

%timeit df.inputs.tolist()
# 100000 loops, best of 3: 11.5 µs per loop