加载文件，.txt文件中的特定行进入数组

Question

I have a .txt file and I need to open it and create an array (in Python). However, I do not want to select all the lines for the array, everything but the first line.

For example, my .txt file reads:

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

and, I would like to create an array such that, I can assign it like:

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

I need to generalize it for future files that will create an array that omits first line of text.

Answer 1

You could do

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

Notice the [1:] which grabs all the other lines except for the first one.

Answer 2

Try:

import numpy

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