Viete, ako získať index alebo stĺpec DataFrame ako pole NumPy alebo zoznam Pythonu?
Ak chcete získať pole NumPy, mali by ste použiť atribút [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)
Tento prístup k údajom je už uložený, takže nie je potrebná konverzia.
Poznámka: Tento atribút je k dispozícii aj pre mnohé iné pandas' objekty.
In [3]: df['A'].values
Out[3]: Out[16]: array([1, 2, 3])
Ak chcete získať index ako zoznam, zavolajte tolist
:
In [4]: df.index.tolist()
Out[4]: ['a', 'b', 'c']
A podobne pre stĺpce.
Od verzie pandas v0.13 môžete použiť aj get_values
:
df.index.get_values()