python - 在 Pandas dataframe.to_csv 中设置缓冲区 0

Question

有没有办法在使用 Pandas dataframe.to_csv() 时将缓冲区设置为“0”？我查看了文档，似乎不允许将其作为参数。我是不是忽略了什么？

编辑:我问是因为我输出的数据帧大小从几百行到几千行不等(总是有相同的 7 列)，而后来最终检查文件的过程偶尔会失败，因为有时它是'写完了。

我当然可以引入延迟(3-5 分钟)，但如果不需要，我宁愿不任意放慢我的代码 - 我宁愿强制代码等待完成在继续之前输出的输出，以及在使用 open() 写入文件时，能够将缓冲区值设置为“0”真是太好了。

Answer 1

如果我没有正确理解您的问题，您可以执行以下操作。此代码段传递了一个 StringIO实例作为 to_csv 的第一个参数, 并调用 seek(0) :

import StringIO

#### your code here...assuming something like:
#### import pandas as pd
#### data = {"key1":"value1"}
#### dataframe = pd.DataFrame(data, index=dataframe)

buffer = StringIO.StringIO()
dataframe.to_csv(buffer)
buffer.seek(0)
output = buffer.getvalue()
buffer.close()

然后您可以根据自己的选择操作输出。