Преобразовать массив string (category) в массив int из массива данных pandas

Я пытаюсь сделать что-то очень похожее на этот предыдущий вопрос, но встречаю ошибку. У меня есть датафрейм pandas, содержащий признаки и метку Мне нужно сделать некоторое преобразование, чтобы отправить признаки и переменную метки в объект машинного обучения:

import pandas
import milk
from scikits.statsmodels.tools import categorical

затем у меня есть:

trainedData=bigdata[bigdata['meta']<15]
untrained=bigdata[bigdata['meta']>=15]
#print trainedData
#extract two columns from trainedData
#convert to numpy array
features=trainedData.ix[:,['ratio','area']].as_matrix(['ratio','area'])
un_features=untrained.ix[:,['ratio','area']].as_matrix(['ratio','area'])
print 'features'
print features[:5]
##label is a string:single, touching,nuclei,dust
print 'labels'

labels=trainedData.ix[:,['type']].as_matrix(['type'])
print labels[:5]
#convert single to 0, touching to 1, nuclei to 2, dusts to 3
#
tmp=categorical(labels,drop=True)
targets=categorical(labels,drop=True).argmax(1)
print targets

Консоль вывода выдает сначала:

features
[[ 0.38846334  0.97681855]
[ 3.8318634   0.5724734 ]
[ 0.67710876  1.01816444]
[ 1.12024943  0.91508699]
[ 7.51749674  1.00156707]]
labels
[[single]
[touching]
[single]
[single]
[nuclei]]

Затем я встречаю следующую ошибку:

Traceback (most recent call last):
File "/home/claire/Applications/ProjetPython/projet particule et objet/karyotyper/DAPI-Trainer02-MILK.py", line 83, in 
tmp=categorical(labels,drop=True)
File "/usr/local/lib/python2.6/dist-packages/scikits.statsmodels-0.3.0rc1-py2.6.egg/scikits/statsmodels/tools/tools.py", line 206, in categorical
tmp_dummy = (tmp_arr[:,None]==data).astype(float)
AttributeError: 'bool' object has no attribute 'astype'

Возможно ли преобразовать переменную категории 'type' в датафрейме в int? 'type' может принимать значения 'single', 'touching', 'nuclei', 'dusts', а мне нужно преобразовать в int значения типа 0, 1, 2, 3.

11
задан Community 23 May 2017 в 12:25
поделиться