基于先前变量的新列

时间:2021-07-09 13:52:45

标签: python pandas

如果第一列中有一个 1,我希望在第二列中有一个 1,如果有不同的值,我希望有一个 0。例如,这就是我所拥有的。

<头>
位置
1
2
4

这就是我想要的:

<头>
位置 新上校
1 1
2 0
4 0

1 个答案:

答案 0 :(得分:0)

你可以用 numpy 来做:

import pandas as pd
import numpy as np

df = pd.DataFrame({
    'Position': [1, 2, 4]
})

df['New col'] = np.array(df['Position'] == 1).astype('int')