缩短DataFrame或缩短CSV文件

Question

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

Answer 1

IIUC，我只会使用head：

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

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