python pandas 如果 A 列中有 1，则使用同一行中 B 列的值

我有两列，BinaryCol 和 OnsetTime 一样，您可能已经猜到了，它是 0 和 1，范围从 0 到 294。我想制作一个新列将包含 OnsetTime 的值，仅适用于 BinaryCol = 1

我目前有这个:

df['Test'] = df['BinaryCol'].apply(lambda row: ['OnsetTime'] if row['BinaryCol'] > 0 else 0, axis=1) 

但它不起作用。

请您参考如下方法:

只做两列的乘积。

df['Test'] = df['OnsetTime'] * df['BinaryCol']