Изменение значений столбца в кадре данных pandas

Question

Я хочу изменить значение столбца Cabin в моем фрейме данных.

Вот мой фрейм данных:

Я хотел бы присвоить значение 1, где Cabin не равно 0. Это означает, что значение C85, C123, E46 должно быть 1, а остальные значения должны быть одинаковыми.

Это код, но я получаю ошибку значения.

if df_train.Cabin != 0:
   df_train.Cabin = 1
else:
   df_train.Cabin = 0

Answer 1

попробуй np.where

df['newCabin'] = np.where(df['cabin'] != 0, 1, df['cabin'])

print(df)
cabin newCabin
    0        0
  C85        1
    0        0
 C123        1
    0        0
    0        0

и если вам не нужен дополнительный новый столбец, вы можете сделать это тоже

df['cabin'] = np.where(df['cabin'] != 0, 1, df['cabin'])
print(df)
cabin
    0
    1
    0
    1
    0
    0

Answer 2

простой oneliner, который должен работать

df_train['cabin'][df_train['cabin'] != '0'] = 1

Answer 3

Просто используйте df.apply:

df_train["new_Cabin"] = df_train["Cabin"].apply(lambda x: x != "0").astype(int)

Или еще проще:

df["new_Cabin"] = (df_train["Cabin"] != "0").astype(int)