Python get_dummiesの例

コード例 #1

def test_only_one_value():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series(["val2"])
    expected = pd.DataFrame([(0, 1, 0)], columns=possible_values)

    actual = get_dummies(data, possible_values)
    assert_frame_equal(expected, actual)

コード例 #2

def test_empty_input():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series([])
    expected = pd.DataFrame([], columns=possible_values)

    actual = get_dummies(data, possible_values)
    assert_frame_equal(expected, actual)

コード例 #3

ファイル: test_get_dummies.py プロジェクト: conorhenley/cincinnati

def test_all_values():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series(["val1", "val3", "val1", "val2"])
    expected = pd.DataFrame([(1, 0, 0), (0, 0, 1), (1, 0, 0), (0, 1, 0)], columns=possible_values)

    actual = get_dummies(data, possible_values)
    assert_frame_equal(expected, actual)

コード例 #4

ファイル: test_get_dummies.py プロジェクト: conorhenley/cincinnati

def test_empty_input():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series([])
    expected = pd.DataFrame([], columns=possible_values)

    actual = get_dummies(data, possible_values)
    assert_frame_equal(expected, actual)

コード例 #5

ファイル: test_get_dummies.py プロジェクト: conorhenley/cincinnati

def test_only_nan():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series([np.nan, np.nan, np.nan, np.nan])
    expected = pd.DataFrame([(np.nan, np.nan, np.nan), (np.nan, np.nan, np.nan), (np.nan, np.nan, np.nan),
                             (np.nan, np.nan, np.nan)], columns=possible_values)

    actual = get_dummies(data, possible_values)
    assert_frame_equal(expected, actual)

コード例 #6

def test_all_values_and_nan():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series(["val1", "val3", "val1", "val2", np.nan])
    expected = pd.DataFrame([(1, 0, 0), (0, 0, 1), (1, 0, 0), (0, 1, 0),
                             (np.nan, np.nan, np.nan)],
                            columns=possible_values)

    actual = get_dummies(data, possible_values)
    assert_frame_equal(expected, actual)

コード例 #7

def test_only_nan():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series([np.nan, np.nan, np.nan, np.nan])
    expected = pd.DataFrame([(np.nan, np.nan, np.nan),
                             (np.nan, np.nan, np.nan),
                             (np.nan, np.nan, np.nan),
                             (np.nan, np.nan, np.nan)],
                            columns=possible_values)

    actual = get_dummies(data, possible_values)
    assert_frame_equal(expected, actual)

コード例 #8

def test_illegal_value():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series(["val1", "val3", "val1", "val4"])

    get_dummies(data, possible_values)

コード例 #9

ファイル: test_get_dummies.py プロジェクト: conorhenley/cincinnati

def test_illegal_value():
    possible_values = ["val1", "val2", "val3"]
    data = pd.Series(["val1", "val3", "val1", "val4"])

    get_dummies(data, possible_values)