DataFrame列添加

Question

我有一个数据帧df，如下所示:(只有一列） 的 col1 0 0 11 34 5 0 72 但我想添加一列col2。如果col1中的元素不为零，则col2中的对应元素将为字符串“play”。但如果它为零，它将为零或只是NA。如下所示。 的 col1 col2 0 NA 0 NA 11 Play 34 Play 5 Play 0 NA 72 Play 提前谢谢。

Answer 1

您可以像这样添加col2：

df['col2'] = np.where(df['col1']!=0, 'Play', 'NA')

Answer 2

尝试：

df['col2'] = df.col1.apply(lambda x: 'Play' if x !=0 else 'NA')