Java FloatArrayをバイト配列に変換してPythonで読み込ませたいです。

Question

Array<Array<FloatArray<>>> (300,300,600) のJava Arrayがあるとします。
この配列を以下のようにbytearrayに変換して、Python側でnp.frombuffer(input,dtype=np.float32)として読み込んだ後に、reshapeで元の(300,300,600)に戻そうと考えています。

# Java
val baos = ByteArrayOutputStream()
val dos =  DataOutputStream(baos)
for (i in 0 until 300){
    for (j in 0 until 300){
        for (k in 0 until 600)
            dos.writeFloat(out1[i][j][k])
       }
   }
}
val out = baos.toByteArray()


#Python
np_array = np.frombuffer(input,dtype=np.float32).reshape(300,300,600)

JavaのFloatArrayを直接numpy配列に変換できる方法もいくつか試しましたが、今回の様に元の配列が大きなサイズだとnp.array()の変換に時間を要したので、他の方法を模索しています。

この方法だと、受け取ったデータが壊れており正しい値を得られません。
どこが間違っているのご指摘願えないでしょうか。
よろしくお願いいたします。

Answer 1

FloatBufferにすることで解決しました。