np.loadtxtで読み込んだデータの形状について

Question

n行5列のデータが入ったテキストファイルをnp.loadtxtを用いて読み込み，読み込んだ値に対して四則演算を行ったのち，その結果をファイルに出力するというプログラムを作成しています．

n=2以上の場合，読み込んだデータの形状は，data.shape=(n,5)となるのですが，n=1の場合は，data.shape=(5,)となってしまいます．
n=1の場合であっても，1行5列の配列（data.shape=(1,5)となるよう）にすることは可能でしょうか．

Answer 1

numpy.loadtxt

ndmin:int, optional

The returned array will have at least ndmin dimensions. Otherwise mono-dimensional axes will be squeezed. Legal values: 0 (default), 1 or 2.

なのでndmin=2を指定するとよいでしょう。

import numpy as np

a = np.array([[1,2,3]])
np.savetxt('num.txt', a)

a = np.loadtxt('num.txt', ndmin=2)
print(a) # [[1. 2. 3.]]

Comments

[nil0]

次元数を指定することができるのですね，教えていただきありがとうございます．
webページも今後の勉強の参考にさせていただきます！