将 pandas DataFrame 更改为 numpy 数组但保留列名

Question

I have a pandas DataFrame from the sklearn.datasets Boston house price data and am trying to convert this to a numpy array but keeping column names. Here is the code I tried:

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)

However this returns None and therefore column names are not kept. Does anyone understand why?

Thanks

Answer 1

try this:

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