python - Pandas 数据框到一维数组

Question

我有一个包含很多列的数据框。

我首先从数据框中只选择一列r_i = df.iloc[:, i: i + 1]

然后我想把这个 r_i 简单地变成数组np.array(r_i).

我想要的结果是这样的:数组([-1，-2，-3])。换句话说，它应该是一个列表的数组。

但是，它给了我一个由子列表组成的列表数组:数组([[-1], [-2], [-3]])。

如何防止这种情况发生？

谢谢。

Answer 1

df.values.flatten()

这里，df 是您的 DataFrame。