使用python读取文件并将值存储在矩阵中

Question

我正在尝试从文件中读取一些数字，并使用Python将它们存储到矩阵中。 在文件的第一行中，我有2个数字n和m，行数和列数，在下一行中，我具有n * m值。 复杂的部分是在文件的第二行中，例如，我没有m值，我只有m-2值。 因此，我无法一次读取一行文件，而只是将值存储在矩阵中。 不能选择编辑文件，因为我有200行1000列的文件。 这是行和列较少的文件的外观：

4 5
1 2 3 
4 5 1 2 3 4 
5 1 2 
3 4 5 1 2 
3 4 5

我设法通过将所有值存储在数组中，然后删除前两个值（即n和m），然后从该数组创建矩阵，来解决了此问题。

这是我的代码：

f = open('somefile2.txt')
numbers = []
for eachLine in f:
    line = eachLine.strip()
    for x in eachLine.split(' '):
        line2 = int(x)
        numbers.append(line2)
f.close()
print numbers
n = numbers[0]
del numbers[0]
m = numbers[0]
del numbers[0]
print n, m, numbers
vector = []
matrix = []
for i in range(n):
    for j in range(m):
        vector.append(numbers[j])
    matrix.append(vector)
    vector = []
print matrix

这给了我预期的结果，但这是通过使用额外的数组numbers来实现的正确方法，还是将所有值直接存储到矩阵中的简便方法？

Answer 1

您可以使用生成器函数：

def solve(f, n, m):
    lis = []
    for line in f:
        if len(lis) > m:
            yield lis[:m]
            lis = lis[m:]
        lis.extend(map(int, line.split()))
    for i in xrange(0, len(lis), m):
        yield lis[i:i+m]       

with open('abc1') as f:
    n, m = map(int, next(f).split())
    # Now you can either load the whole array at once using the list() call,
    # or use a simple iteration to get one row at a time.
    matrix = list(solve(f, n, m))
    print matrix

输出：

[[1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5]]

---

另一种方法是获取文件中所有项目的扁平化迭代器，然后将该迭代器分成大小相等的块。

from itertools import chain, islice

with open('abc1') as f:
    n, m = map(int, next(f).split())
    data = chain.from_iterable(map(int, line.split()) for line in f)
    matrix = [list(islice(data, m)) for i in xrange(n)]
    print matrix
    #[[1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5]]

---

有关：

• 如何在Python中将列表分成大小均匀的块？

• 在Python中展平浅表

Answer 2

我的2美分：

with open('somefile.txt') as f:
    strings = f.read().split()

numbers = map(int, strings)
m = numbers.pop(0)
n = numbers.pop(0)

matrix = [numbers[i:i+n] for i in xrange(0, m*n, n)]

在Python 3中，您只需执行以下操作：

m, n, *numbers = map(int, strings)

根据您要处理的数据，您可能需要看一下NumPy，它有一些不错的读取文本文件的方法。