以下是pandas DataFrame示例:
id product_type qty
1 product_type 1 100
2 product_type 2 300
3 product_type 1 200
我想删除product_type
列中的product_type
,以获取以下新的DataFrame:
id product_type qty
1 1 100
2 2 300
3 1 200
这就是我尝试这样做的方式:
orders['product_type'].strip('product_type ')
但是有一个错误:
'Series' object has no attribute 'strip'
答案 0 :(得分:5)
你需要.str
,因为它是string accessor method:
orders['product_type'].str.strip('product_type ')
In [6]:
df['product_type'] = df['product_type'].str.strip('product_type ')
df
Out[6]:
id product_type qty
0 1 1 100
1 2 2 300
2 3 1 200
或传递正则表达式将数字提取到str.extract
:
In [8]:
df['product_type'] = df['product_type'].str.extract(r'(\d+)')
df
Out[8]:
id product_type qty
0 1 1 100
1 2 2 300
2 3 1 200