def make_date_columns_categorical_binary(book_attributes):
"""Turn all date columns in book_attributes into binary categorical columns."""
# bucket publish dates & insert categorical data columns into data frame
orig_pub_year_cat = transform_pub_dates(book_attributes['original_pub_year'])
book_attributes.insert(loc=5, column='orig_pub_year_cat', value=orig_pub_year_cat)
pub_year_cat = transform_pub_dates(book_attributes['pub_year'])
book_attributes.insert(loc=5, column='pub_year_cat', value=pub_year_cat)
# turn date categories into binary dataframes; merge back into book_attributes
pub_year_dummies = pd.get_dummies(book_attributes['pub_year_cat'])
orig_year_dummies = pd.get_dummies(book_attributes['orig_pub_year_cat'])
book_full_attr = book_attributes.merge(pub_year_dummies,left_index=True, right_index=True)
book_full_attr = book_full_attr.merge(orig_year_dummies,left_index=True, right_index=True)
return book_full_attr
recommendation_creation.py 文件源码
python
阅读 25
收藏 0
点赞 0
评论 0
评论列表
文章目录