加载文件,从.text文件到数组的特定行

时间:2013-10-07 01:27:18

标签: python arrays file-io

我有一个.txt文件,我需要打开它并创建一个数组(在Python中)。但是,我不想为数组选择所有行,除了第一行之外的所有行。

例如,我的.txt文件显示为:

 1 1 1
 1 4 6
 4 5 6
 8 9 7

并且,我想创建一个数组,我可以像下面这样分配它:

 Y= array([[1, 4, 6], [4, 5, 6], [8, 9, 7]])

我需要将它概括为将来创建一个省略第一行文本的数组的文件。

2 个答案:

答案 0 :(得分:0)

你可以做到

with open(file) as f:
    Y = [map(int, line.split()) for line in f.readlines()[1:]]

注意[1:]抓取除第一行之外的所有其他行。

答案 1 :(得分:0)

尝试:

import numpy

data = numpy.loadtxt('data.txt',skiprows=1)