Замена значений в столбцах значениями из других столбцов в соответствии с отображением ⇐ Python

[Гость]

У меня есть такой вид данных: < /p>

Код: Выделить всё

df = pd.DataFrame({
"A1": [1, 11, 111],
"A2": [2, 22, 222],
"A3": [3, 33, 333],
"A4": [4, 44, 444],
"A5": [5, 55, 555]
})

A1   A2   A3   A4   A5
0    1    2    3    4    5
1   11   22   33   44   55
2  111  222  333  444  555
< /code>
и этот вид отображения: < /p>
mapping = {
"A1": ["A2", "A3"],
"A4": ["A5"]
}
< /code>
Что означает, что я хочу, чтобы все столбцы в списке имели значения из столбца ключа, поэтому: A2 и A3 должны быть заполнены значениями из A1, а A5 должен быть заполнен значениями из A4. Полученный DataFrame должен выглядеть следующим образом: < /p>
    A1   A2   A3   A4   A5
0    1    1    1    4    4
1   11   11   11   44   44
2  111  111  111  444  444
< /code>
Мне удалось сделать это довольно просто так: < /p>
for k, v in mapping.items():
for col in v:
df[col] = df[k]

Но мне было интересно, есть ли векторизованный способ сделать это (более пандактический способ)?


Подробнее здесь: https://stackoverflow.com/questions/796 ... -to-mappin