【NumPy】CSV形式でファイル保存

この記事では、Python言語とNumPyを用いて、CSV形式でファイル保存する方法をソースコード付きで解説します。

CSV形式の保存

NumPyでは、savetxtメソッドでNumPy配列内のデータをファイル出力できます。

【書式】

numpy.savetxt(テキストファイル名, ndarray, delimiter=区切り文字)

※CSVファイルで保存する場合、区切り文字にはカンマ(,)を入れます。

ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8
import numpy as np

A = np.array([[1, 2],
              [3, 4],
              [5, 6]])

np.savetxt('data1.csv', A, delimiter=",")
np.savetxt('data2.csv', A.T, delimiter=",")

NumPy配列をそのままsavetxtメソッドに入れると縦に並べて保存されます。
NumPy配列を.Tで転置してからsavetxtメソッドに入れると横に並べて保存されます。

実行結果

サンプルプログラムの実行結果です。

data1.csv

1.000000000000000000e+00,2.000000000000000000e+00
3.000000000000000000e+00,4.000000000000000000e+00
5.000000000000000000e+00,6.000000000000000000e+00

data2.csv

1.000000000000000000e+00,3.000000000000000000e+00,5.000000000000000000e+00
2.000000000000000000e+00,4.000000000000000000e+00,6.000000000000000000e+00