你知道如何将 DataFrame 的索引或列作为 NumPy 数组或 python 列表获取吗?
要获取 NumPy 数组,您应该使用以下values属性:
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:
tolist
In [4]: df.index.tolist() Out[4]: ['a', 'b', 'c']
同样,对于列。