cumsum和小数的问题



假设以下代码:

import numpy as np
import pandas as pd

df = pd.DataFrame({'A': np.random.randint(0, 10, 10)})
df['B'] = df['A'].diff()
x, x_diff = 1, df['B'].iloc[1:]
df['C'] = np.r_[x, x_diff].cumsum()
A    B    C
# 0  6  NaN  1.0
# 1  6  0.0  1.0
# 2  0 -6.0 -5.0
# 3  7  7.0  2.0
# 4  5 -2.0  0.0
# 5  3 -2.0 -2.0
# 6  3  0.0 -2.0
# 7  8  5.0  3.0
# 8  8  0.0  3.0
# 9  8  0.0  3.0

C列的变化正如预期的那样非常漂亮。这似乎很顺利。然而,当我使用十进制数字时,它们会四舍五入到0,最终我的起始值根本没有改变。有什么办法防止这种情况发生吗?理论上我可以再次将数字相乘,但有更好的方法来解决这个问题吗?问题如下所示:

import pandas as pd
import numpy as np
df = pd.DataFrame({'A': np.random.randint(0, 10, 10)})/100000000000000

df['B'] = df['A'].diff()
x, x_diff = 1, df['B'].iloc[1:]
df['C'] = np.r_[x, x_diff].cumsum()
#               A             B    C
# 0  9.000000e-14           NaN  1.0
# 1  7.000000e-14 -2.000000e-14  1.0
# 2  1.000000e-14 -6.000000e-14  1.0
# 3  9.000000e-14  8.000000e-14  1.0
# 4  9.000000e-14  0.000000e+00  1.0
# 5  4.000000e-14 -5.000000e-14  1.0
# 6  6.000000e-14  2.000000e-14  1.0
# 7  9.000000e-14  3.000000e-14  1.0
# 8  7.000000e-14 -2.000000e-14  1.0
# 9  0.000000e+00 -7.000000e-14  1.0

感谢您的众多建议。使用0作为PUFF建议的起始值是有效的。Chris提出的CCD_ 1方法同样有效。

最新更新