熊猫字符串到数字

Question

我有一组类似的数据

        a      b
0  type 1   True
1  type 2  False

如何保留a 列的数字部分并同时将 ture 转换为 1，将 false 转换为 0。下面是我想要的。

   a  b
0  1  1
1  2  0

Answer 1

您可以将布尔值转换为整数，如下所示：

df['b'] = df.b.astype(int)

根据A 列中文本的性质，您可以做一些事情：

a) 在空间上拆分并获取第二部分（字符串或 int 取决于您的需要）。

df['a'] = df.a.str.split().str[1]  # Optional `.astype(int)`

b) 使用正则表达式从字符串末尾提取任何数字 (\d*)。

df['a'] = df.a.str.extract(r'(\d*)$')  # Optional `.astype(int)`