Как в pandas сделать список на основании данных нескольких столбцов?

Question

san_m_m @san_m_m

Python

Как в pandas сделать список на основании данных нескольких столбцов?

Добрый день!

Есть DataFrame

import pandas as pd
df = pd.DataFrame({'movie': [9999999,  2,  3, 1, 9999999],
                  'rating': [3,  2,  9999999, 9999999, 3],
                  'name': [1,  2,  4, 5, 10]})

Нужно добавить еще один столбец, в котором будут данные по следующей логике...
Если в столбце movie 9999999, то берется значение из столбца rating, если в rating 9999999, то name
попробовал написать следующий код, но как мне кажется идея в корне не верна, поэтому я тут...

kol = []
for l in df:
    for i in l:
        if i['movie'] != 9999999:
            k = i['movie']
        elif i['rating'] != 9999999:
            k = i['rating']
        else:
            k = i['name']
        kol.append(k)
df['sum'] = kol

Вопрос задан более трёх лет назад
137 просмотров

1 комментарий

Подписаться 2 Простой 1 комментарий

Вам что, для задания выдали мало букв? Как насчёт поименовать переменные внятно, чтоб было понятно в чем суть всех этих l, i…

Answer 1 · 2021-07-29 13:09:37

Можно вынести вычисления в функцию и применить apply.

import pandas as pd
df = pd.DataFrame({'movie': [9999999,  2,  3, 1, 9999999],
                  'rating': [3,  2,  9999999, 9999999, 3],
                  'name': [1,  2,  4, 5, 10]})

def process(row):
    result = row['movie']
    if row['movie'] == 9999999:
        if row['rating'] == 9999999:
            result = row['name']
        else:
            result = row['rating']
    return result

df['sum'] = df.apply(process, axis=1)

Как в pandas сделать список на основании данных нескольких столбцов?

Войдите на сайт