Панды - замена значений в определенном столбце

Question

Я пытаюсь заменить все значения в определенном столбце в кадре данных, если он больше 0. Я пробовал ниже, но он заменяет все столбцы этой конкретной строки.

df.loc[(df['count'] > 0)] = 1

Может кто-нибудь помочь, спасибо..

Answer 1

Вы близки, нужно указать столбец во втором параметре:

df.loc[df['count'] > 0, 'count'] = 1

Вы также можете указать несколько столбцов:

df.loc[df['count'] > 0, ['count', 'another col']] = 1

Другое решение:

df['count'] = np.where(df['count'] > 0, 1, df['count'])
#alternative    
#df['count'] = df['count'].mask(df['count'] > 0, 1)

Answer 2

попробуйте следующий пример кода;это более гибко:

import pandas as pd
import numpy as np

data = np.random.randint(low=-2, high=3, size=10)
df = pd.DataFrame(data=data, columns=['value'])
df['new_value'] = df.value.apply(lambda i : i if i < 0 else 1)
print(df)

Результат:

Answer 3

df.loc[(df['count'] > 0), 'count'] = 1

Я рекомендую этот урок о .loc https://www.youtube.com/watch?v=xvpNA7bC8cs