熊猫-如何将所有非零值映射到列中的1

Question

我有一个像这样的表

data = pd.DataFrame({'one': ['A', 'B', 'C', 'D'], 'two': [1, 0, 2, 0]})

我想将所有非0的值更改为1。

我创建了一个循环：

for i in range(data.shape[0]):
    if data.iloc[i, 1] == 0:
        pass
    else:
        data.iloc[i, 1] = 1

但是我正在寻找一些熊猫函数（例如map）

谢谢，马特