如何使用Pandas从循环向数据框追加行？

Question

我正在运行一个循环，将值附加到循环的空数据框外。 但是，完成此操作后，数据帧仍为空。 我不确定发生了什么。 目标是找到导致残差平方和最小的功率值。

示例代码如下：

import tweedie

power_list = np.arange(1.3, 2, .01)
mean = 353.77
std = 17298.24
size = 860310
x = tweedie.tweedie(mu = mean, p = 1.5, phi = 50).rvs(len(x))
variance = 299228898.89

sum_ssr_df = pd.DataFrame(columns = ['power', 'dispersion', 'ssr'])

for i in power_list:

    power = i

    phi = variance/(mean**power)

    tvs = tweedie.tweedie(mu = mean, p = power, phi = phi).rvs(len(x))

    sort_tvs = np.sort(tvs)

    df = pd.DataFrame([x, sort_tvs]).transpose()
    df.columns = ['actual', 'random']
    df['residual'] = df['actual'] - df['random']
    ssr = df['residual']**2
    sum_ssr = np.sum(ssr)
    df_i = pd.DataFrame([i, phi, sum_ssr])
    df_i = df_i.transpose()
    df_i.columns = ['power', 'dispersion', 'ssr']
    sum_ssr_df.append(df_i)    

sum_ssr_df[sum_ssr_df['ssr'] == sum_ssr_df['ssr'].min()]

我究竟做错了什么？

Answer 1

这段代码的效率不如ALollz所说的那么高效。 当你追加时，它基本上会在内存中创建一个新的数据帧（我在这里过于简化了）。

您的代码中的错误是：

 sum_ssr_df.append(df_i)

应该：

 sum_ssr_df = sum_ssr_df.append(df_i)