【数据挖掘】多特征组合的基本方法

  • Post author:
  • Post category:其他


举个相当简单的例子:在一个csv文件中(已经通过pandas读入为data),需要通过服装的颜色、风格、销量和类别去预测其价格。每个特征分别属于一列,现在我们要组合(已经预处理好的)特征,有如下的写法:

  1. 先选出特征对应的列名,再得到所有选取的特征
features_select = ['color', 'style', 'sale', 'category']
combine_features = data[features_select]

features_dummies = pd.get_dummies(combine_features)
  1. 直接组合不同特征列
combine_features = pd.concat([data['color'], data['style'], data['sale'], data['category']], axis=1)

features_dummies = pd.get_dummies(combine_features)
  • 相比来说,

    pd.concat()更加灵活

    ,方便更自由的组合特征列,需要注意的是:concat函数中将

    具体的

    特征数据放入[ ]里,最后axis=1代表

    行拼接

  • 而第一种方式则需要一次性将特征的

    列名

    列举完整,如果特征并不是完全在同一个文件中,会稍微麻烦一些。

接上面的例子,比如特征中的sale已经是一个具体的int值了,所以我不需要对其进行get_dummies处理,这个时候就可以先组合其他需要get_dummies的特征,最后再concat上sale即可:

features_select = ['color', 'style', 'category']
features_dummies = pd.get_dummies(data[features_select])
features_final = pd.concat([features_dummies, data['sale], axis=1)

还有刚刚说的如果采用第一种方式时不是所有的特征都在data中,还有特征在store文件中(比如store_num)。此时可以直接在程序的运行过程中将store指定列拉取到data中,如下:

data = pd.read_csv(...)
store = pd.read_csv(...)
上面的不解释,就是默认已经读入了两个文件
data['store_num_new'] = data['store_num']
这一步实际上是在程序运行过程中给data植入了一新列,实际csv没有改变
然后很自然地将store_num_new当作data的一列就好啦
(concat组合 列名组合均可)

上面只是一个小例子,其实本质上仅仅是不同的代码写法而已,本质上是相同的。实际情况中是根据具体的数据处理需求,

结合两种写法

对特征进行灵活的处理和组合。


多踩bug,多进步 ~ 从bug中学习,也希望最终能成功做出项目!



版权声明:本文为qq_16763983原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。