python - 用数组替换整个系列值

Question

我有一个看起来像这样的系列：

13624    A
25257    D
15359    D
26226    D
10797    D
        ..
9733     D
25121    D
10143    C
1448     B
13627    B
Name: points, Length: 10189, dtype: object

我想应用LabelEncoder()函数sklearn.preprocessing并替换所有值。

该le.fit_transform(y_train)函数返回：

array([0, 3, 3, ..., 2, 1, 1])

这个数组是按原始索引排序的，所以想法是用这个数组完全替换值列。

我怎样才能做到这一点？

标签： pythonpandas

Answer 1

您可以尝试创建具有相同索引的新系列并重新分配：

y_train = pd.Series(le.fit_transform(y_train), index=y_train.index)