Python 用tabpy预测性别数的结果

我有一个由datetime索引和男女计数组成的年度数据，我在tableau计算字段中创建了一个模型，下面是代码

SCRIPT_REAL(
"
import pandas as pd
import numpy as np
from statsmodels.tsa.arima.model import ARIMA
from random import random

period = _arg3[0]+1

model = ARIMA(_arg2[:-period], order=(2,1,2))
model_fit= model.fit()
future = model_fit.predict(1,len(_arg2), typ='levels')

print(_arg2[:-period])

return future.tolist()

",ATTR([Date]),[Provider Count],[Forecast Year])

我得到了计数的预测，但是当我尝试使用性别过滤器时，我确实得到了结果，但是男性预测和女性预测的总和不等于总计数的预测。当我使用filter Gender='M'时，我怀疑发生了什么，然后只传递男性计数数据，并预测结果，女性和总计数也发生了相同的情况。我想要的是使用total count预测模型，当我使用过滤器gender='M'时，total count的预测结果应用于显示男性和女性计数