Handle Infinite Values and Division by Zero in Pandas

Question

When dividing revenue by ad spend where spend is 0, Pandas produces `inf`. Detect all infinite values in the DataFrame and replace them with `NaN`.

Accepted Answer

Optimal Solution:
import numpy as np
import pandas as pd

df['roas'] = df['revenue'] / df['ad_spend']

# Replace inf and -inf with NaN
df['roas'] = df['roas'].replace([np.inf, -np.inf], np.nan)

result_df = df[['roas']]

Technical Explanation:
`np.inf` values are replaced with `np.nan` using `.replace([np.inf, -np.inf], np.nan)`, making them compatible with standard `fillna()` or `dropna()`.

revenue	ad_spend
1000	0
5000	1000

roas
NULL
5