python出前三行数据 python怎样取数据的前几行

以读取并保存txt文件前三列数据为例：先打开文件，再读取文件，最后保存前三列数据。

import codecs

# 读取并保存txt文件数据前三列数据
f = codecs.open('XXX.txt', mode='r', encoding='utf-8')  # 打开txt文件，以‘utf-8'编码读取
line = f.readline()      # 以行的形式进行读取文件
list = []
while line:
    a = line.split(',')  # 当数据以“，”分隔时，用line.split(',')
    b = a[0:3]           # 选取需要读取数据的列数
    list.append(b)       # 将数据添加在list中
    line = f.readline()
# 保存前三列数据
f = open (r'XXX_1.txt','w')
for i in list:
    print (i,file = f)   # 保存数据到XXX_1.txt
f.close()