缩短 DataFrame 或缩短 CSV 文件

Question

我正在从 csv 文件读取数据、进行计算并将数据返回到不同的 csv 文件。我有一个来自原始文件的 100 行长 8 列的 DataFrame。就地数据计算后，100 行数据被“压缩”，因此现在我只想将 90 行数据返回到下游 csv 文件。接收 csv 文件的系统不喜欢接收最后 10 行被全零覆盖的数据文件，因此我想找到一种仅在数据帧中写入前 90 行的方法。有没有办法做到这一点？从我所见，“to_csv”没有内置字段用于将少于整个数据帧发送到 csv 文件。

Answer 1

IIUC，我只会使用head：

df.head(90).to_csv('yourfile.csv')

通过这种方式，您可以将数据帧开头的第一行n 写入文件。