python - 你应该如何读取二维整数数组？

Question

我是 python 的新手。我想从文本文件中读取一个二维整数数组，文件中每行一行，数字用逗号分隔。这有效:

s = f.read()
ss = s.splitlines()
mx = []
for i in range(0,len(ss)):
    mx.append([])
    for s1 in ss[i].split(','):
        mx[i].append(int(s1))

有没有更简单的方法来做到这一点？

Answer 1

有一个用于读取逗号分隔文件的内置模块 ( csv ):

import csv
mx = [map(int, row) for row in csv.reader(open(filename))]

另外，如果您打算用这个数组做数学运算，您可能会发现安装和使用 numpy 很有用。 :

import numpy
mx = numpy.loadtxt(filename, delimiter=',', dtype=int)

mx 现在将是 array而不是列表的列表。