使用熊猫从CSV文件读取特定行的功能

Question

我有一个包含200行的CSV文件。我想创建一个函数来读取每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')

这样，您就不需要遍历数据框。