使用numpy读取列数不等的数据文件

Question

我有一个。带有数字的dat文件。在第一行中，该文件有五列，在所有后续行中，它有四列。我希望能够使用numpy读取此文件。我目前尝试读取此文件时遇到以下错误：

In [3]: F1 = np.loadtxt(‘file.dat')
---------------------------------------------------------------------------
ValueError                                Traceback (most recent 
call last)
<ipython-input-3-c0f31adaf29a> in <module>()
----> 1 F1 = np.loadtxt(‘file.dat')

/Users/usr/anaconda2/lib/python2.7/site-packages/numpy/lib/npyio.pyc in loadtxt(fname, dtype, comments, delimiter, converters, skiprows, usecols, unpack, ndmin, encoding)
   1090         # converting the data
   1091         X = None
-> 1092         for x in read_data(_loadtxt_chunksize):
   1093             if X is None:
   1094                 X = np.array(x, dtype)

/Users/usr/anaconda2/lib/python2.7/site-packages/numpy/lib/npyio.pyc in read_data(chunk_size)
   1014                 line_num = i + skiprows + 1
   1015                 raise ValueError("Wrong number of columns at line %d"
-> 1016                                  % line_num)
   1017 
   1018             # Convert each value according to its column and store

ValueError: Wrong number of columns at line 2 

如何使用python读取文件中除第一行之外的所有行？我在这里附上了一个示例文件。

此外，该文件的第一列（减去第一行）有n^2行（在示例中，我有n=3，该列的条目为1,2,3,4,5,6,7,8,9）。我想读取第一列（减去第一行）并将其保存为文本文件，其中形状为（n，n）（即文本文件应该有n行和n列）。也就是说，我希望保存的矩阵具有以下顺序的条目：

1.0 2.0 3.0
4.0 5.0 6.0
7.0 8.0 9.0

我很感激能得到帮助。

Answer 1

要做的一些实验：（未优化）1.读入文件的行：

编辑：file.dat文件有空行。ifline.strip（）...子句用于处理空行。

with open('file.dat', 'r') as fhand:
    file_lines = [line[:-1] for line in fhand if line.strip() != ''] # remove the last character '\n'. **Remove empty lines**.

如果你不喜欢第一排，就把它放下。

file_lines.pop(0)

现在，剩余的行具有相同数量的数字条目列，您可以拆分每行中的条目，并进行类型转换：

mat_raw = [[float(term) for term in line.split()] for line in file_lines]

然后你得到一个浮点矩阵。为了方便切片，将其转换为numpy数组。

mat = numpy.array(mat_raw)
# then you can do whatever you like. eg: first column
first_col = mat[:, 0]
# reshape it to n by n matrix:
res = first_col.reshape((n, n))
...

根据文件格式和目标，您可以优化此代码以供自己使用。