本文整理汇总了Python中zipline.lib.labelarray.LabelArray.as_categorical_frame方法的典型用法代码示例。如果您正苦于以下问题:Python LabelArray.as_categorical_frame方法的具体用法?Python LabelArray.as_categorical_frame怎么用?Python LabelArray.as_categorical_frame使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类zipline.lib.labelarray.LabelArray
的用法示例。
在下文中一共展示了LabelArray.as_categorical_frame方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_string_classifiers_produce_categoricals
# 需要导入模块: from zipline.lib.labelarray import LabelArray [as 别名]
# 或者: from zipline.lib.labelarray.LabelArray import as_categorical_frame [as 别名]
def test_string_classifiers_produce_categoricals(self):
"""
Test that string-based classifiers produce pandas categoricals as their
outputs.
"""
col = TestingDataSet.categorical_col
pipe = Pipeline(columns={'c': col.latest})
run_dates = self.trading_days[-10:]
start_date, end_date = run_dates[[0, -1]]
result = self.run_pipeline(pipe, start_date, end_date)
assert isinstance(result.c.values, Categorical)
expected_raw_data = self.raw_expected_values(
col,
start_date,
end_date,
)
expected_labels = LabelArray(expected_raw_data, col.missing_value)
expected_final_result = expected_labels.as_categorical_frame(
index=run_dates,
columns=self.asset_finder.retrieve_all(self.asset_finder.sids),
)
assert_frame_equal(result.c.unstack(), expected_final_result)