区分字母数字和在Python中创建新列

问题描述

数据框如下所示： df：

    item_id value 
0   101002  1.008665
1   101004  2.025818
2   101005  0.530516
3   101007  0.732918
4   101010  1.787264
... ... ...
621 ZB005   3.464102
622 ZB007   2.345208
623 ZB008   3.464102
624 ZBD002  2.592055
625 ZBD005  2.373321

我想基于列newcol的第一个元素/字母创建一个新列item_id ： 如果item_id的第一个字母是字母，则返回“字母”；如果第一个字母是数字，则返回“数字”。

预期输出：

    item_id value     newcol
0   101002  1.008665  number
1   101004  2.025818  number 
2   101005  0.530516  number
3   101007  0.732918  number
4   101010  1.787264  number
... ... ...
621 ZB005   3.464102  alphabet 
622 ZB007   2.345208  alphabet 
623 ZB008   3.464102  alphabet 
624 ZBD002  2.592055  alphabet 
625 ZBD005  2.373321  alphabet 

我尝试过：

df['new_component'] = [lambda x: 'alphabet' if x.isalpha() else 'number' for x in df.item_id]

返回了

    item_id value       new_component
0   101002  1.008665    <function <listcomp>.<lambda> at 0x000002663B04E948>
1   101004  2.025818    <function <listcomp>.<lambda> at 0x000002663B04E828>
2   101005  0.530516    <function <listcomp>.<lambda> at 0x000002663B04EAF8>
3   101007  0.732918    <function <listcomp>.<lambda> at 0x000002663B04EB88>
4   101010  1.787264    <function <listcomp>.<lambda> at 0x000002663B04EC18>
... ... ...

代码有什么问题？有更好的方法吗？

解决方法

首先将列设置为默认值'alphabet'，然后将其更改为数字：

df['newcol'] = 'alphabet'
df.loc[df.item_id.str[0].str.isdigit(),'newcol'] = 'number'

如果您喜欢尝试使用的方法，请按以下步骤操作：

df['newcol'] = ['number' if x[0].isdigit() else 'alphabet' for x in df.item_id]

或等效地：

df['newcol'] = ['alphabet' if x[0].isalpha() else 'number' for x in df.item_id]

输出：

    item_id     value    newcol
0    101002  1.008665    number
1    101004  2.025818    number
2    101005  0.530516    number
3    101007  0.732918    number
4    101010  1.787264    number
621   ZB005  3.464102  alphabet
622   ZB007  2.345208  alphabet
623   ZB008  3.464102  alphabet
624  ZBD002  2.592055  alphabet
625  ZBD005  2.373321  alphabet

,

在列表理解中，您正在创建lambda函数的列表。

您需要做的就是首先在列表之外定义lambda函数

lhs

然后，在列表理解中调用它。