1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
--- a/README.rst
+++ b/README.rst
@@ -229,7 +229,7 @@
array([[2015, 10, 30],
[2015, 10, 31],
[2015, 11, 1],
- [2015, 11, 2]])
+ [2015, 11, 2]], dtype=int32)
We can also specify this option per group of columns instead of for the
whole mapper::
@@ -241,7 +241,7 @@
array([[2015, 10, 30],
[2015, 10, 31],
[2015, 11, 1],
- [2015, 11, 2]])
+ [2015, 11, 2]], dtype=int32)
Outputting a dataframe
**********************
--- a/tests/test_dataframe_mapper.py
+++ b/tests/test_dataframe_mapper.py
@@ -340,8 +340,8 @@
transformed = mapper.fit_transform(df)
cols = transformed.columns
assert len(cols) == 4
- assert cols[0] == 'target_x0_0'
- assert cols[3] == 'target_x0_3'
+ assert cols[0] == 'target_0'
+ assert cols[3] == 'target_3'
def test_customtransform_df():
@@ -917,8 +917,8 @@
columns = sorted(list(outdf.columns))
assert len(columns) == 2
- assert columns[0] == 'colA_a'
- assert columns[1] == 'colA_b'
+ assert columns[0] == 'colA_0'
+ assert columns[1] == 'colA_1'
def test_with_car_dataframe(cars_dataframe):
|
