带有多行数据的Numpy genfromtxt

Question

我有一个文本文件，每个记录都是一个4x3的数值块，例如

1 1 1
1.0 1.0 1.0
1.0 1.0 1.0
1.0 1.0 1.0

2 2 2
2.0 2.0 2.0
2.0 2.0 2.0
2.0 2.0 2.0

我试图将它们读入一个numpy结构化数组。 genfromtext和loadtxt似乎都希望将数据表示为单行而不是供应商提供的自定义格式。 我知道可以通过标准python读取行，将每4行的组推入单个字符串，然后将字符串传递给numpy，但我想知道numpy是否本身支持这种类型的操作？ 我在genfromtxt docs中没有看到任何内容。

例如， genfromtxt(fname, skip_header=1, shape=(4,3), dtype=my_custom_dtype)

Answer 1

我只是将您的数据放入文件中并测试了loadtxt 。 它会将所有内容读取到一个数组中，然后只需要拆分即可。

a = np.loadtxt(fname)
res = np.vsplit(a, a.shape[0]/4)