如何根据 Python Pandas 中的其他列在 DataFrame 中创建新列？ [复制]

Question

我有如下数据框：

data = pd.DataFrame({"col1" : ["a", "b", "c"], 
                     "binary" : [0, 1, 0]})

我想在这个名为"new" 的DataFrame 中创建并添加新列，如果在"binary" 列中为1，那么我想在我创建的列"new" 中获取来自"col1" 的数据，但如果在"binary" 列中为 0，那么我想要np.NaN。以下是我根据上面的描述和 DatFrame 真正需要的结果。

Answer 1

您可以使用 numpy 包中的 where 和 , 并执行以下操作：

import pandas as pd
import numpy as np
df = pd.DataFrame({"col1" : ["a", "b", "c"], "binary" : [0, 1, 0]})
df['new'] = np.where(df['binary']==1,df['col1'],np.nan)

prints:

print(df)
  col1  binary  new
0    a       0  NaN
1    b       1    b
2    c       0  NaN