如何将pandas Series或index转换为Numpy数组？

Question

如何将pandas Series或index转换为Numpy数组？您知道如何将DataFrame的索引或列作为NumPy数组或python列表获取吗？

Answer 1

要获取NumPy数组，您应该使用以下values属性：

In [1]: df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]}, index=['a', 'b', 'c']); df

   A  B

a  1  4

b  2  5

c  3  6

In [2]: df.index.values

Out[2]: array(['a', 'b', 'c'], dtype=object)

这可以访问数据的存储方式，因此无需进行转换。

注意：此属性也可用于许多其他pandas的对象。

In [3]: df['A'].values

Out[3]: Out[16]: array([1, 2, 3])

要将索引作为列表，请调用tolist：

In [4]: df.index.tolist()

Out[4]: ['a', 'b']

同样，对于列。

Answer 2

您可以使用df.index访问索引对象，然后使用列表中的值df.index.tolist()。同样，您可以使用df['col'].tolist()系列。