我使用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,因为似乎一切都在工作,直到脚本的最后。
2条答案
按热度按时间lh80um4z1#
这在很大程度上取决于您所应用的功能。它返回一个dataframe还是一个series?
icnyk63a2#
如果只有一个列,Groupby将产生一个系列(假设有内容),所以使用pandas数据框将不再起作用
在编写csv之前,请尝试执行以下操作
df2 = df2.reset_index()
然后写csv
reset_index()从序列中重建pandas数据范围