使用loc对数据库进行排序时，数据文件只保存12行

Question

我正在尝试在 pandas 中使用 .loc 迭代地选择 500 行和所有列。但是，最终写入的文件最多只能显示 12 行数据。前 5 行和后 5 行。 写入文件中不显示中间数据。需要做什么才能显示所有 500 行???

以下是代码

"""
Program to 
1. load data from a file. Skip the first 52 rows of data from a tab-separated file. 
2. Select 500 rows and all columns from the dataset. Write this file. 
3. Use the next name for saving the next set of 500 rows and iterate this process for 20 times

"""

import pandas as pd
import numpy as np
import os
import datetime
import shutil
import math
import datetime

b='b'
b=pd.read_csv('Oct4_3.dat',delimiter='\t',skiprows=52)
c=0
d=500


for i in range (1,20):
    a='a'
    filename = 'a'+str(i)+'.dat'
        
    a=a+str(i)
    print(a)    

    a=b.loc[c:d,'# V UG']   

    f = open(filename,'a+')    
    f.write(" %s \n"%(a))
    c=d+1
    d=d+500
    f.close()
   
print('done')

代码输出生成 20 个文件。一个特定的文件如下所示

Answer 1

您在文件中获得的内容似乎是 print(a) 的结果。 要将 pandas 数据帧保存为 csv，请使用 to_csv() 方法

a.to_csv('your_csv_filename.csv',index=False)

此外，您正在为第一次迭代保存 501 行而不是 500 行（索引从 0 开始而不是 1）