Измените pandas DataFrame на numpy массив, но сохраните имена столбцов

Question

У меня есть pandas DataFrame из данных о ценах на дома в Бостоне sklearn.datasets, и я пытаюсь преобразовать его в массив numpy, но сохраняя имена столбцов. Вот код, который я пробовал:

from sklearn import datasets ## imports datasets from scikit-learn
import numpy as np
import pandas as pd

data = datasets.load_boston() ## loads Boston dataset from datasets library

df = pd.DataFrame(data.data, columns=data.feature_names)
X = df.to_numpy()
print(X.dtype.names)

Однако это возвращает None, и поэтому имена столбцов не сохраняются. Кто-нибудь понимает почему?

Спасибо

Answer 1

попробуйте это:

w = (data.feature_names).reshape(13,1)
X = np.vstack((w.T, data.data))
print (X)