Python ColumnSelector.get_feature_names 예제들

예제 #1

def test_get_feature_names():
    df = pd.DataFrame({'a': [4, 5, 6], 'b': ['4', '5', '6']})
    transformer = ColumnSelector('a').fit(df)
    assert transformer.get_feature_names() == ['a']

예제 #2

파일: test_columnselector.py 프로젝트: prateekchandrajha/scikit-lego

def test_get_feature_names():
    df = pd.DataFrame({"a": [4, 5, 6], "b": ["4", "5", "6"]})
    transformer = ColumnSelector("a").fit(df)
    assert transformer.get_feature_names() == ["a"]