关于 python：如何用计算的 CAGR 值替换 NaN 列

  val1  val2  val3  val4  val5
0 100    100   100   100  100
1  90    110    80   110   50
2  70    150    70   NaN   NaN
3  NaN    NaN   NaN  NaN   NaN

from __future__ import division # for python2.7
import numpy as np

# tab delimited data
a = ”’100  100 100 100 100
90  110 80  110 50
70  150 70  NaN NaN
NaN NaN NaN NaN NaN
”’

# parse and make a numpy array
data = np.array( [[np.nan if aaa==‘NaN’ else int(aaa) for aaa in aa.split(‘//t’)] for aa in a.splitlines()] )

for col in range(5):
   
    Nyears = np.isnan(data[:,col]).argmax()–1 # row index for the last non-NaN value
    endvalue = data[Nyears,col]
    cagr = (endvalue / 100) ** (1 / Nyears)
    print Nyears, endvalue, cagr

    for year in np.argwhere(np.isnan(data[:,col])):
        data[year,col] = data[year-1,col] * cagr

print data

[[ 100.          100.          100.          100.          100.        ]
 [  90.          110.           80.          110.           50.        ]
 [  70.          150.           70.          121.           25.        ]
 [  58.56620186  183.71173071   58.56620186  133.1          12.5       ]]