BUG: AssertionError calling df.shift(axis=1) on empty DataFrame · Issue #57301 · pandas-dev/pandas (original) (raw)

Pandas version checks

• I have checked that this issue has not already been reported.
• I have confirmed this bug exists on the latest version of pandas.
• I have confirmed this bug exists on the main branch of pandas.

Reproducible Example

import pandas as pd pd.DataFrame().shift(axis=1)

Issue Description

The above code raises an AssertionError:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Users/joshi/miniconda3/envs/pandas-latest/lib/python3.11/site-packages/pandas/core/frame.py", line 5799, in shift
    return super().shift(
           ^^^^^^^^^^^^^^
  File "/Users/joshi/miniconda3/envs/pandas-latest/lib/python3.11/site-packages/pandas/core/generic.py", line 10706, in shift
    assert axis == 0  # axis == 1 cases handled in DataFrame.shift
           ^^^^^^^^^
AssertionError

Expected Behavior

Shifting an empty dataframe should return an empty dataframe, regardless of axis (this behavior apparently worked in pandas 1.5.3).

Installed Versions

INSTALLED VERSIONS

commit : 7fe86b6
python : 3.11.7.final.0
python-bits : 64
OS : Darwin
OS-release : 23.3.0
Version : Darwin Kernel Version 23.3.0: Wed Dec 20 21:31:00 PST 2023; root:xnu-10002.81.5~7/RELEASE_ARM64_T6020
machine : arm64
processor : arm
byteorder : little
LC_ALL : None
LANG : en_US.UTF-8
LOCALE : en_US.UTF-8

pandas : 3.0.0.dev0+263.g7fe86b67d3
numpy : 2.0.0.dev0+git20240203.5c34c87
pytz : 2024.1
dateutil : 2.8.2
setuptools : 68.2.2
pip : 23.3.1
Cython : None
pytest : None
hypothesis : None
sphinx : None
blosc : None
feather : None
xlsxwriter : None
lxml.etree : None
html5lib : None
pymysql : None
psycopg2 : None
jinja2 : None
IPython : None
pandas_datareader : None
adbc-driver-postgresql: None
adbc-driver-sqlite : None
bs4 : None
bottleneck : None
dataframe-api-compat : None
fastparquet : None
fsspec : None
gcsfs : None
matplotlib : None
numba : None
numexpr : None
odfpy : None
openpyxl : None
pandas_gbq : None
pyarrow : None
pyreadstat : None
python-calamine : None
pyxlsb : None
s3fs : None
scipy : None
sqlalchemy : None
tables : None
tabulate : None
xarray : None
xlrd : None
zstandard : None
tzdata : 2023.4
qtpy : None