我Python 2.7在Mac OSX Lion上使用。我使用的是IPython与Pandas 0.11.0,Numpy和Statsmodels包。
我正在编写一个函数,该函数允许用户在文件上进行逻辑回归,指定要在构建模型中使用的变量,应将哪些变量转换为虚拟变量,以及哪些变量应为自变量。
例如,当我执行以下操作时:
cols_to_keep = []
print (df.columns)
i = eval(raw_input('How many of these variables would you like to use in logistic regression?: '))
while i != 0:
i = i - 1
print (df.columns)
addTo = raw_input('Enter a variable for this list that you would like to keep and use in logistic regression.: ')
cols_to_keep.append(addTo)
我最终遇到了麻烦。具体来说,当我要求用户从列表中指定因变量,然后需要将该变量从训练变量列表中删除时:
print (df.columns)
dependent = raw_input('Which of these columns would you like to be the dependent variable?: ')
training.remove(dependent)
在插入打印语句后,我发现添加到训练变量列表中的变量看起来像这样:
('these are the traing variables: ', ['access', u'age_age6574', u'age_age75plus', u'sex_male', u'stage_late', u'death_death'])
似乎u已经在每个用户指定的变量之前放置了。
我的问题是:这是为什么,以及如何解决/解决此问题,以便当用户指定因变量时,实际上已将其从列表中删除。在用户指定变量并将其添加到列表的所有其他情况下,也会发生这种情况,如果我需要用户观察列表,则会造成混乱。
HUX布斯
相关分类