如何在Pandas DataFrame中计算IRR？解决一维数组输入报错问题

错误原因 #

npf.irr() 函数要求输入一维数组（rank-1 array），即单个现金流序列。你直接传入 df.iloc[:,1:] 是一个二维DataFrame（多行多列），不符合函数输入要求。此外，Rate1/Rate2/Rate3列用逗号作为小数分隔符，属于字符串类型，无法直接参与数值计算，这也是潜在问题。

分步解决方案 #

1. 转换小数格式为数值类型 #

先将含逗号的字符串列转换为浮点型：

import pandas as pd
import numpy_financial as npf

# 替换逗号为小数点并转为数值
df[['Rate1', 'Rate2', 'Rate3']] = df[['Rate1', 'Rate2', 'Rate3']].replace(',', '.', regex=True).astype(float)

2. 逐行计算IRR #

使用apply()方法对每一行（axis=1）调用npf.irr()，确保每行传入的是一维数组：

# 对每行的现金流列（Amount_paid到Rate3）计算IRR
df['IRR'] = df.iloc[:, 1:].apply(npf.irr, axis=1)

最终结果示例 #

处理后的DataFrame会新增IRR列，包含每行的内部收益率计算结果：

注意事项 #

如果某行的现金流序列无有效IRR（例如全为正或全为负），npf.irr()会返回NaN或抛出异常，可添加错误处理优化：

def calculate_irr(row):
    try:
        return npf.irr(row)
    except:
        return np.nan

df['IRR'] = df.iloc[:, 1:].apply(calculate_irr, axis=1)

内容的提问来源于stack exchange，提问作者Olga Małecka