python替换非na值[重复]

Question

我想创建一个新列，并用0和1替换NA和不丢失的值。

#df
col1  
1      
3

5

6

我想要的：

#df
col1   NewCol
1           1
3           1
            0
5           1
            0
6           1

这就是我所尝试的：

df['NewCol']=df['col1'].fillna(0)
df['NewCol']=df['col1'].replace(df['col1'].notnull(), 1)

看来第二行不正确。
有什么建议吗？

Answer 1

首先，您需要将所有的na转换为0。如何做到这一点将因范围而异。对于单个列，您可以使用：

df['DataFrame Column'] = df['DataFrame Column'].fillna(0)

对于整个数据帧，您可以使用：

df.fillna(0)

在此之后，需要将所有非零替换为“1”。您可以这样做：

for index, entry in enumerate(df['col']):
    if entry != 0:
        df['col'][index] = 1

请注意，此方法将0计为空条目，这可能是也可能不是所需的功能。

Answer 2

您可以尝试：

df['NewCol'] = [*map(int, pd.notnull(df.col1))]

希望这有帮助。