python - Pandas 仅从数据框中选择数字或整数字段

Question

我有这个 Pandas 数据框 (df):

     A    B
0    1    green
1    2    red
2    s    blue
3    3    yellow
4    b    black

类型是对象。

我会选择 A 值为整数或数字的记录:

     A    B
0    1    green
1    2    red
3    3    yellow

谢谢

Answer 1

在数据帧上调用 apply(注意双方括号 df[['A']] 而不是 df['A']) 并调用字符串方法 isdigit()，然后我们设置参数 axis=1 以逐行应用 lambda 函数。这里发生的是索引用于创建 bool 掩码。

In [66]:
df[df[['A']].apply(lambda x: x[0].isdigit(), axis=1)]
Out[66]:
       A       B
Index           
0      1   green
1      2     red
3      3  yellow

更新

如果您使用的版本是 0.16.0或更高版本则以下内容也将起作用:

In [6]:
df[df['A'].astype(str).str.isdigit()]

Out[6]:
   A       B
0  1   green
1  2     red
3  3  yellow

这里我们使用 astype 将 Series 转换为 str，然后调用向量化的 str.isdigit

另请注意，convert_objects 已弃用，对于最新版本 0.17.0 或更高版本，应使用 to_numeric