DataFrame のインデックスまたは列を NumPy 配列または Python リストとして取得するにはどうすればよいですか?
ベストアンサー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', 'c']
列についても同様です。