如何删除df中整个列的某个字符后的所有内容？

Question

I have a df with one column (SKUID) where I want to remove all the characters that are not numerical. Here is an sample of the column:

Essentially I want to remove the underscore and the letter for each row. I have tried using following code:

sku_data.split('_', 1)[0]

This gives me an error of 'DataFrame' object has no attribute 'split'. Where am I going wrong?

Answer 1

This should do for number extraction:

sku_data.SKUID = sku_data.SKUID.str.extract('(\d+)')

Note : don't forget to add the str operator if you want to perform string operations on a DataFrame column