スポンサーリンク

numpyでcsvを読み込みmatplotlibでグラフを作成

csv形式のファイルを読み込み後、

npArray = np.loadtxt("in.txt", delimiter = ",")

xとy成分をスライスで指定します。
xは0列目、yは2列目を指定しています。
参考:Pythonのnumpy配列をスライスで行と列の取得をする方法

x = npArray[:, 0]
y = npArray[:, 2]

折れ線グラフを作成します。

plt.plot(x, y)
plt.show()

例えば、下記のin.txtがあったとします。

0,1,8,3
4,5,6,7
8,9,10,11
10,9,10,11

下記がサンプルコードになります。

$ cat sample.py 
#!/usr/bin/env python3
# coding: UTF-8

import numpy as np
import matplotlib.pyplot as plt

npArray = np.loadtxt("in.txt", delimiter = ",")

x = npArray[:, 0]
y = npArray[:, 2]

plt.plot(x, y)
plt.show()

下記が実行結果になります。
$ ./sample.py 

numpy-csv-plot

スポンサーリンク