разбить данные на куски - PullRequest
1 голос
/ 25 марта 2020

Возможно, этот вопрос прост, но я пытаюсь найти способ, как это сделать автоматически c, предположим, у меня есть фрейм данных, и я хочу разбить его на куски и сохранить их, используя имена на основе кусок. Мне удалось как разделить, но как я могу сохранить его в .cvs, используя имена data_1, data_2 и так далее ... 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) ]

1 Ответ

1 голос
/ 25 марта 2020

Вы можете использовать встроенный метод 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, et c .. .

...