Description
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 numpy as np
import pandas as pd
x = pd.Series([np.nan, 0.0, 1.2, pd.NA]).convert_dtypes()
y = pd.Series([np.nan, 0.0, 30.2, 10]).convert_dtypes()
z = pd.Series([np.nan, 0.0, 15.2, 9.2]).convert_dtypes()
(z**2 / (x*y)).convert_dtypes()Issue Description
numpy.nan resulting from an arithmetic operation (e.g., division by zero) is not being converted to pd.NA. The above examples outputs:
0 <NA>
1 NaN
2 6.375276
3 <NA>
dtype: Float64
Expected Behavior
I expect all np.nan present in the Series are converted to pd.NA:
0 <NA>
1 <NA>
2 6.375276
3 <NA>
dtype: Float64
Installed Versions
INSTALLED VERSIONS
commit : 139def2
python : 3.12.3
python-bits : 64
OS : Linux
OS-release : 6.8.0-41-generic
Version : #41-Ubuntu SMP PREEMPT_DYNAMIC Fri Aug 2 20:41:06 UTC 2024
machine : x86_64
processor : x86_64
byteorder : little
LC_ALL : None
LANG : en_GB.UTF-8
LOCALE : en_GB.UTF-8
pandas : 3.0.0.dev0+1545.g139def2145
numpy : 2.2.0.dev0+git20240930.3ee9e6a
dateutil : 2.9.0.post0
pip : 24.0
Cython : None
sphinx : None
IPython : None
adbc-driver-postgresql: None
adbc-driver-sqlite : None
bs4 : None
blosc : None
bottleneck : None
fastparquet : None
fsspec : None
html5lib : None
hypothesis : None
gcsfs : None
jinja2 : None
lxml.etree : None
matplotlib : None
numba : None
numexpr : None
odfpy : None
openpyxl : None
psycopg2 : None
pymysql : None
pyarrow : None
pyreadstat : None
pytest : None
python-calamine : None
pytz : None
pyxlsb : None
s3fs : None
scipy : None
sqlalchemy : None
tables : None
tabulate : None
xarray : None
xlrd : None
xlsxwriter : None
zstandard : None
tzdata : 2024.2
qtpy : None
pyqt5 : None