将数据分成块

Question

也许这个问题很简单，但我正在尝试找到一种方法如何以自动方式完成它，假设我有一个数据框并且我想将它分成块并使用基于块的名称保存它们.我管理了如何拆分，但如何使用名称 data_1、data_2 等将其保存在 .cvs 中... PS。我只需要块，所以我在这里没有使用 sklearn 方法。

import numpy as np
import pandas as pd
df = pd.DataFrame({
    'x_values':np.random.randn(400),
    'y_values':np.random.randn(400),
})

sample = 100
N = int(len(df)/sample)
frames = [df.iloc[i*sample:(i+1)*sample] for i in range(N+1) ]

Answer 1

您可以使用to_csv内置方法：

[df.iloc[i*sample:(i+1)*sample].to_csv('data_'+str(i)+'.csv') for i in range(N+1) ]

这将创建名为 data_0.csv、data_1.csv 等的单独数据框...

将数据分成块

split data into chunks

python

chunks