帶有多行數據的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)