python - 有没有一种简单的方法可以将 Pandas 数据框中的 yes/no 列更改为 1/0？

Question

我将 csv 文件读入 pandas 数据帧，并希望将具有二进制答案的列从是/否字符串转换为 1/0 整数。下面，我展示了其中一列(“sampleDF”是 pandas 数据框)。

In [13]: sampleDF.housing[0:10]
Out[13]:
0     no
1     no
2    yes
3     no
4     no
5     no
6     no
7     no
8    yes
9    yes
Name: housing, dtype: object

非常感谢您的帮助!

Answer 1

方法一

sample.housing.eq('yes').mul(1)

方法2

pd.Series(np.where(sample.housing.values == 'yes', 1, 0),
          sample.index)

方法3

sample.housing.map(dict(yes=1, no=0))

方法4

pd.Series(map(lambda x: dict(yes=1, no=0)[x],
              sample.housing.values.tolist()), sample.index)

方法5

pd.Series(np.searchsorted(['no', 'yes'], sample.housing.values), sample.index)

---

全部 yield

0    0
1    0
2    1
3    0
4    0
5    0
6    0
7    0
8    1
9    1

---

时机
给定样本

时机
长样
sample = pd.DataFrame(dict(housing=np.random.choice(('yes', 'no'), size=100000)))