python - 只写从 pandas df 到 csv 的前 N ​​行

Question

如何仅将前 N 行或从 P 到 Q 行从 pandas 数据帧写入 csv 而无需先对 df 进行子集化？由于内存问题，我无法对要导出的数据进行子集化。

我正在考虑一个逐行写入csv的函数。

谢谢

标签： pythonpandascsv

Answer 1

• 使用head - 返回前 n 行。

前任。

import pandas as pd
import numpy as np
date = pd.date_range('20190101',periods=6)
df = pd.DataFrame(np.random.randn(6,4), index=date, columns=list('ABCD'))

#wtire only top two rows into csv file
print(df.head(2).to_csv("test.csv"))