Pandas groupby()+ apply()创建csv

wqnecbli  于 2023-06-20  发布在  其他
关注(0)|答案(2)|浏览(126)

我使用pandas groupby()+apply()从另一个脚本运行一个函数,并从输出创建一个csv。它似乎一直工作到最后,当将dataframe写入csv时,它返回一个没有任何内容的csv。
我导入另一个脚本中编写的函数,并读取输入csv:

from FILE import FUNCTION

df= pd.read_csv('/path/input.csv')

然后创建一个空的dataframe,使用groupby()+ apply()将函数应用于组,然后写入新的csv:

df2 = []
df2 = df.groupby("column.1").apply(FUNCTION)
# #
# #
# #
df2.to_csv('/path/new.csv', sep=',', index=False)

我不明白为什么它返回一个空的csv,因为似乎一切都在工作,直到脚本的最后。

lh80um4z

lh80um4z1#

这在很大程度上取决于您所应用的功能。它返回一个dataframe还是一个series?

icnyk63a

icnyk63a2#

如果只有一个列,Groupby将产生一个系列(假设有内容),所以使用pandas数据框将不再起作用
在编写csv之前,请尝试执行以下操作
df2 = df2.reset_index()
然后写csv
reset_index()从序列中重建pandas数据范围

相关问题