文字列を含むnp.savetxtの保存方法

Question

現在データセットを作成するためにデータセットのpath, 分類結果1, 分類結果2といったshapeが(318,3)のnumpy配列を作成し、この配列をcsv形式に保存しようと

np.savetxt("all_train.csv", all_train)

を用いて保存を試みたのですが、

*** TypeError: Mismatch between array dtype ('<U23') and format specifier ('%.18e %.18e %.18e')

といったエラーメッセージが返ってきました。感覚的にはunicodeの型がダメだったのかと感じ

all_train.astype(np.str)

などを使って型を変えようかとも思いましたが変わらなかったです。
どなたかわかる方がいらっしゃいましたらご教授をお願い致します。

Answer 1

fmt="%s"を使うと文字列として出力できます。
が、文字メインのデータを扱うのであればpandasを使う方が楽かと思います。

import numpy as np

all_train = np.array([['path/to/hoge','OK','NG']])
print(all_train.shape)

np.savetxt("all_train.csv", all_train,  fmt="%s")

a = np.loadtxt("all_train.csv",dtype="str")
print(a)

Comments

[chutora]

ありがとうございます！
pandasで読み込んだデータの欠損値を整形してsavetxtで保存するといった算段だったのですが、その場合、dtypeがobjectのデータを保存することになりますけど同じようにfmtを%sにすればいけるんですかね？(すみません、自分でやってみます)

Answer 2

型がデフォルトでは、%.18e と浮動小数点型となっているため、文字列の場合は明示的に指定する必要があります。

a = [
    ["test.txt", "OK", "NG"],
    ["test.txt", "OK", "NG"],
    ["test.txt", "OK", "NG"],
    ["test.txt", "OK", "NG"],
]

import numpy as np

np.savetxt("test.txt", a, fmt="%s")

Comments

[chutora]

なるほど…！fmtの変更も一応試みたのですが、fmt=‘str’と公式にも書いてないようなことをしたのがまずかったみたいです…。
ともあれ解決できました、ありがとうございます！