【问题标题】:How to replace column values by matching values in a dictionary of a different length in Python?如何通过匹配Python中不同长度的字典中的值来替换列值?
【发布时间】:2023-04-06 15:08:02
【问题描述】:

我有一个看起来像这样的数据框

ID   Color
A    red
B    green
C    red
D    yellow

我通过创建字典将颜色枚举为数字:

color_list = ['red', 'green', 'yellow']
colors = dict(enumerate(color_list))

现在我如何用颜色 ID 替换列值,以便数据框如下所示:

ID  Color
A    1
B    2
C    1
D    3

编辑:作为后续问题,如果我在 Spark RDD 中有相同的数据,我将如何在 Scala 中处理它?

【问题讨论】:

    标签:
    python
    scala
    pandas
    dictionary
    apache-spark