python - 替换列中的值

Question

这是我的数据集，其中有四列。我想将 survival_status 列中的值(1 和 2)替换为(负数和正数)。我正在使用 pandas 来更改值。

 Age  operation_year  axillary_nodes_detected  survival_status
0   30              64                        1                1
1   30              62                        3                1
2   30              65                        0                2
3   31              59                        2                1
4   31              65                        4                2

Haberman["survival_status"] = Haberman["survival_status"].apply(lambda x : 'Positive' if x == 2 else 'Negative')

应用后，它将整个列值更改为负值。

Haberman['survival_status'].value_counts()
Negative    306
Name: survival_status, dtype: int64

谁能告诉我我哪里做错了？

Answer 1

一种方法是使用字典映射。但首先请确保您的数据帧已转换为 int:

df = df.astype(int)

d = {2: 'Positive', 1: 'Negative'}

df['survival_status'] = df['survival_status'].map(d)

结果:

print(df)

   Age  operation_year  axillary_nodes_detected survival_status
0   30              64                        1        Negative
1   30              62                        3        Negative
2   30              65                        0        Positive
3   31              59                        2        Negative
4   31              65                        4        Positive