在循环中存储和添加 Pandas 数据帧

Question

由于我的Generate_Dataframe函数，我有一个循环，每次都会生成随机结果。 因此，名称列保持不变，但我的"结果"列始终是不同的浮点值。

def Generate_Dataframe():
 #Do Some Stuff
  return DataFrame  

我回来的一个例子是这样的...

DataFrame
Name 1       Name 2        Result
Joe          Smith           5.5
Jake         Smith           4.5
Jim          Smith           2.5
Joanne       Smith           1.5

因此，当我运行循环时，我正在生成如上面的数据帧，我希望每次都能够更新/添加结果列。

for x in range(1,5):
    New_DataFrame = Generate_DataFrame()

我无法找到存储数据帧的方法。 优化速度会很棒。 提前感谢！

Answer 1

IIUC 您正在使用索引等name列。 您应该将它们放在那里，然后更新/添加变得微不足道。

def gen_df():
    midx = pd.MultiIndex.from_tuples([
        ('Joe', 'Smith'),
        ('Jake', 'Smith'),
        ('Jim', 'Smith'),
        ('Joanne', 'Smith')
    ], names=['Name 1', 'Name 2'])
    return pd.DataFrame(
        dict(Result=np.random.rand(4)),
        midx
    )

---

选项 1
您不必这样做。 但这就是我会这样做的：

from functools import reduce
reduce(pd.DataFrame.add, (gen_df() for _ in range(1, 5)))
                 Result
Name 1 Name 2          
Joe    Smith   2.400550
Jake   Smith   2.222812
Jim    Smith   2.601639
Joanne Smith   0.503774

---

选项 2
在循环中

df = gen_df()
for _ in range(1, 5):
    df += gen_df()
df
                 Result
Name 1 Name 2          
Joe    Smith   1.998055
Jake   Smith   2.268697
Jim    Smith   2.815204
Joanne Smith   2.253301

Answer 2

如果你想存储数据帧，我认为最好的方法是保存到一个pickle/csv文件。df.to_pickle(file_name(/df.to_csv(file_name(

您可以阅读：如何使用 Pandas 存储数据帧