Skip to content

stack() casts to float when called on levels not requiring it #17886

Closed
@toobaz

Description

Code Sample, a copy-pastable example if possible

In [2]: df = pd.DataFrame(0, index=range(3),
   ...:                   columns=pd.MultiIndex.from_tuples([('w', 'a', 'j'),
   ...:                                                      ('w', 'a', 'k'),
   ...:                                                      ('w', 'b', 'j')],
   ...:                                                     names=['v', 'x', 'y']))

In [3]: df.stack(['x', 'y'])
Out[3]: 
v        w
  x y     
0 a j  0.0
    k  0.0
  b j  0.0
1 a j  0.0
    k  0.0
  b j  0.0
2 a j  0.0
    k  0.0
  b j  0.0

Problem description

Casting to float is not required (notice it would have been if df.stack() had been called without arguments).

Expected Output

In [4]: df.stack(['x', 'y']).astype(int)
Out[4]: 
v      w
  x y   
0 a j  0
    k  0
  b j  0
1 a j  0
    k  0
  b j  0
2 a j  0
    k  0
  b j  0

Output of pd.show_versions()

INSTALLED VERSIONS

commit: 6793a136dde139d968948b6ec9193cc3535abf6f
python: 3.5.3.final.0
python-bits: 64
OS: Linux
OS-release: 4.9.0-3-amd64
machine: x86_64
processor:
byteorder: little
LC_ALL: None
LANG: it_IT.UTF-8
LOCALE: it_IT.UTF-8

pandas: 0.21.0rc1+13.g6793a136d.dirty
pytest: 3.0.6
pip: 9.0.1
setuptools: None
Cython: 0.25.2
numpy: 1.12.1
scipy: 0.19.0
pyarrow: None
xarray: None
IPython: 5.1.0.dev
sphinx: 1.5.6
patsy: 0.4.1
dateutil: 2.6.0
pytz: 2017.2
blosc: None
bottleneck: 1.2.1
tables: 3.3.0
numexpr: 2.6.1
feather: 0.3.1
matplotlib: 2.0.0
openpyxl: None
xlrd: 1.0.0
xlwt: 1.1.2
xlsxwriter: 0.9.6
lxml: None
bs4: 4.5.3
html5lib: 0.999999999
sqlalchemy: 1.0.15
pymysql: None
psycopg2: None
jinja2: 2.9.6
s3fs: None
fastparquet: None
pandas_gbq: None
pandas_datareader: 0.2.1

Metadata

Assignees

No one assigned

    Labels

    BugDtype ConversionsUnexpected or buggy dtype conversionsReshapingConcat, Merge/Join, Stack/Unstack, Explode

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions