Функция для чтения указанных c строк из файла CSV с pandas

Question

У меня есть файл CSV, содержащий 200 строк. Я хочу создать функцию для чтения каждые 50 строк вместе, а затем сохранить их (50 строк) в файле .txt, пока не закончится файл CSV. Как я могу сделать это, пожалуйста? Любая помощь приветствуется.

import pandas as pd
import csv

def my_function(n):
  dataset = pd.read_csv('e.csv', nrows=50) 
  X = dataset.iloc[:,[0,0]].values

Обновление ::

def my_function(n):

dataset = pd.read_csv('e.csv', nrows=n) 
X = dataset.iloc[:,[0,0]].values


with open('funct.txt', 'w') as file:
    for i in X:
        file.write("{}\n".format(i))

return

row_count = len(open("e.csv").readlines())
print(row_count)
n=50

my_function(n)

Теперь моя проблема как можно читать каждые 50 строк за другой в каждом время до достижения максимальной длины (200)?

Answer 1

Вы можете проверить оставшуюся часть евклидова деления индекса на 50, чтобы проверить, является ли ваш номер строки кратным 50:

df=pd.read_csv('e.csv')
df=df[df.index%50==0]
df.to_csv('newfile.txt')

Таким образом, вам не нужно перебирать ваш фрейм данных.