Pandas:一列除以另一列得到两列NaN

import pandas as pd

# 创建示例数据
data = {'column1': [1, 2, 0, 4, 5],
        'column2': [0, 2, 3, 0, 5]}
df = pd.DataFrame(data)

# 计算结果
df['result1'] = df['column1'] / df['column2']
df['result2'] = df['column2'] / df['column1']

print(df)

   column1  column2  result1   result2
0        1        0      inf  0.000000
1        2        2      1.0  1.000000
2        0        3      0.0       inf
3        4        0      inf  0.000000
4        5        5      1.0  1.000000

df = df.fillna(float('nan'))
print(df)

   column1  column2  result1  result2
0        1        0      NaN      0.0
1        2        2      1.0      1.0
2        0        3      0.0      NaN
3        4        0      NaN      0.0
4        5        5      1.0      1.0