python - 将 Pandas 数据集转换为数组，以便在 Scikit-Learn 中建模

Question

我们能否在 Pandas DataFrames 上运行 scikit-learn 模型，或者我们是否需要将 DataFrames 转换为 NumPy 数组？

Answer 1

您可以将 pandas.DataFrame 与 sklearn 一起使用，例如:

import pandas as pd
from sklearn.cluster import KMeans

data = [(0.2, 10),
        (0.3, 12),
        (0.24, 14),
        (0.8, 30),
        (0.9, 32),
        (0.85, 33.3),
        (0.91, 31),
        (0.1, 15),
        (-0.23, 45)]

p_df = pd.DataFrame(data)
kmeans = KMeans(init='k-means++', n_clusters=3, n_init=10)
kmeans.fit(p_df)

结果:

>>> kmeans.labels_
array([0, 0, 0, 2, 2, 2, 2, 0, 1], dtype=int32)