如何从文本文件读取数组数据到numpy数组？

Question

我有一个文本文件data.txt，其数据格式如下：

 [[ 1.0   2.0   3.0]
 [1.0    2.0   3.0]
 [1.0    2.0   3.0]
 [1.0    2.0   3.0]]

如何将这种格式的数据读入numpy中的jupyter数组中？

Answer 1

import numpy as np

with open("data.txt") as infile:
  my_array = np.array([map(float,line.strip(" []\n").split()) for line in infile.readlines()])

Answer 2

这应该可行，并且可以推广到float以外的其他类型：

with open("data.txt") as infile:
    np.fromstring( infile.read().replace("[","").replace("]", ""), sep="   ").reshape(-1,3)

注意：np.fromstring返回一个1d数组，我添加了一个重塑，假设有3列。