如何解决区分字母数字和在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然后，在列表理解中调用它。
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
