【发布时间】:2021-04-23 14:05:22
【问题描述】:
我有一个带有行 ID 的表:
| ID | fruit | place |
|---|---|---|
| 61725RN | apple | London |
| 12381YH | banana | Toronto |
我需要列出一个只有 ID 的数字部分的列表:['61725', '12381']
我编写下面的代码并收到“KeyError”。但是,列名在我的代码中是粗体的,所以它是一个索引。
nums = df['ID'].str[:5]
【问题讨论】:
-
尝试
df.reset_index()之后尝试df['ID'].str[:5] -
请提供完整的错误信息,但我猜测列名中有前导或尾随空格:
'ID '检查df.columns以查看