python - 在 Python 中重新编码变量，类似于 R ifelse

Question

我只是 Python 的新手，我正在尝试进行分析。但在此之前，我想重新编码一些变量。我想知道这个 R 代码在 Python 中是否有等价物。

df$col1 <- ifelse(df$col1 == "yes", 1, 0)

这df是一个pandas.DataFrame和col1是它的列之一。

标签： pythonrdataframenumpyrecode

Answer 1

一种可能的选择 - 可能在语法方面与 R 最相似ifelse- 是使用where来自的函数numpy

import numpy as np

df['col1'] = np.where(df['col1'] == "Yes", 1, 0)