Как я могу создать функцию, которая показывает, кто-то сделал комментарий или нет?

Question

Я использую этот набор данных из kaggle .

Существует столбец, в котором показано, какие респонденты оставили комментарий, а какие нет.Респонденты, которые не комментируют, имеют значение NaN.Я хочу создать другую переменную, которая показывает 1, если кто-то сделал комментарий, и 0, если кто-то не сделал комментарий.Это код, который я использовал:

for i in df['comments']:
    if i==np.nan:
        return 0
    elif i!=np.nan:
        return 1

Однако я получаю следующую ошибку:

return за пределами функции.

Любойпомощь будет высоко ценится.

Answer 1

Попробуйте, это вам поможет !!!

import numpy as np

def fun():
    m = 0 if i is np.nan else 1
    return m

for i in data["comments"]:
    new = fun()
    print(new)