Заменить все значения в pandas фрейме данных поиском из другого фрейма данных - PullRequest
1 голос
/ 07 марта 2020

Учитывая следующие данные:

df = pd.DataFrame(
    dict(
        x1=["zero", "one", "two"],
        x2=["one", "zero", "zero"],
        x3=["zero", "two", "one"],
        x4=["zero", "two", "two"],
    )
)

, который выглядит как:

In [2]: df
Out[2]:
     x1    x2    x3    x4
0  zero   one  zero  zero
1   one  zero   two   two
2   two  zero   one   two

Я хотел бы заменить элементы в нем, используя следующие:

reps = pd.DataFrame(
    dict(
        val_from=["zero", "one", "two"],
        val_to=["nothing", "single", "couple"],
    )
)

в результате получаются следующие данные:

    x1    : x2     : x3     : x4
  single  : single : single : single
   single : single : couple : couple
   couple : single : single : couple

Следующее работает, но я чувствую, что есть лучший подход:

replacement = {x:y for x,y in zip(reps['val_from'], reps['val_to'])}
df.transform(lambda x: x.replace(replacement))

1 Ответ

2 голосов
/ 07 марта 2020

Вы можете использовать DataFrame.replace, передавая Series с индексом "from" и значениями "to" из reps:

df.replace(reps.set_index(['val_from'])['val_to'])

[out]

        x1       x2       x3       x4
0  nothing   single  nothing  nothing
1   single  nothing   couple   couple
2   couple  nothing   single   couple
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...