Pythonでテキストの読み書き (NumPy)

numpy.loadtxt

テキストデータをnumpy arrayに読み込む 詳しい記述はソースサイトより

よく使うオプション

• fname: file, string
• dtype: data-type, optional
• delimiter: string, optional

numpy.savetxt

numpy arrayをテキストデータに書き出す

よく使うオプション

• fname: file, string
• X: 1D or 2D array_like
• fmt: str or sequence of strs, optional
• delimiter: string, optional
• header: str, optional
• footer: str, optional

例

osとargparseと組み合わせて連番ファイル (カンマ区切り) を読み込む。

import argparse
parser = argparse.ArgumentParser(description="""""")
parser.add_argument("--version", action="version", version='%(prog)s 0.0.1')
parser.add_argument("-i", "--inputdir", metavar="input-dir", default="./", help="input dir")
parser.add_argument(      "--time-start", default=1, type=int, help="start number")
parser.add_argument(      "--time-nstep", default=1, type=int, help="end number")
options = parser.parse_args()

import numpy
import os

for n, i in enumerate(range(options.time_start, options.time_nstep+1)):
  fname = os.path.join(options.rdir, "data.{0:04d}.txt".format(i))
  data = numpy.loadtxt(fname, ftype, delimiter=",")