将对象列转换为数字

Question

I am currently working with a python dataframe and there is a names column with different names in the form of strings.

I try to convert these names into individual numbers so for example:

"Paul" 1
"John" 2
"Nina" 3
...

I already tried

dataset['name'] = dataset['name'].astype(np.int64)

and this:

dataset = dataset.astype({'collection_author':'int'})

Which both lead to the following error:

ValueError: invalid literal for int() with base 10

and also this:

dataset['name'] = pd.to_numeric(dataset['name'])

which leads to:

ValueError: Unable to parse string "John" at position 0

If I add the errors='coerce' the code runs through however most of the values are replaced with NaN.

Answer 1

使用pandas.get_dummies或sklearn.preprocessing.OneHotEncoder将字符串转换为整数。