python:以numpy.loadtxt格式读取文本文件,用数字分割整数

时间:2016-07-28 19:55:55

标签: python

我的程序读取格式如下的文本文件,每个数字之间有空格:

0 1 1
1 0 0
1 0 0

我目前阅读文本文件的代码是

G = numpy.loadtxt(filename, int)

如果我打印(G),输出如下:

[[0 1 1]
 [1 0 0]
 [1 0 0]]

我收到了我需要在我的程序上运行的新txt文件,但文本文件在每个数字之间没有空格,如下所示:

011
100
100 

我希望能够将这些新的txt文件读入2D列表,而不像以前那样使用逗号。我试过这个:

filename = open(file, "r")
G = []
gr = filename.readline().strip()
while gr:
    gr = list(map(int,str(gr)))
    G.append(gr)
    gr = filename.readline().strip()

当我打印(G)时,它看起来像这样,并不适用于我的程序:

[[0, 1, 1], [1, 0, 1], [1, 1, 0]]

有没有办法读取这些新文本文件,每个数字之间没有空格,并且格式与之前的格式相同?

1 个答案:

答案 0 :(得分:1)

你已经拥有的几乎正在工作。只需在代码底部添加一行,如下所示:

G = numpy.array(G)