Python中DataFrame分类数据转换报错：'DataFrame'无属性'i'的解决方法

错误原因 #

你写的dfTrain.i是尝试访问DataFrame名为i的属性，但这里i是循环变量，代表列表里的列名，应该用方括号索引dfTrain[i]来动态获取对应列。.只能访问固定名称的列，无法用变量代替，所以才会触发'DataFrame' object has no attribute 'i'的错误。

修正后的代码 #

import pandas as pd

CATEGORICAL_COLUMNS = ['sex','n_siblings_spouses', 'parch', 'class',
               'embark_town', 'alone']
for i in CATEGORICAL_COLUMNS:
  dfTrain[i] = pd.factorize(dfTrain[i])[0]
dfTrain.head()

补充优化方案 #

如果想更简洁地批量处理，可以用apply方法：

dfTrain[CATEGORICAL_COLUMNS] = dfTrain[CATEGORICAL_COLUMNS].apply(lambda x: pd.factorize(x)[0])

内容的提问来源于stack exchange，提问作者Cole Hendrickson