PythonでCSVファイルを読み込み、グラフを作成する方法

Pythonの標準モジュールである「csvモジュール」やデータ解析支援ライブラリ「pandas」を用いて、CSVファイルからデータを読み込み、グラフを作成する方法について紹介します。

CSVファイルの読み込みとグラフ作成

まずは、Pythonの標準モジュールである「csvモジュール」を用いてCSVデータを読み取り、matplotlibでグラフ化する方法について紹介します。

import csv
import numpy as np
import matplotlib.pyplot as plt

path_csv = r'test\\test.csv'
rows = []

with open(path_csv) as f:
    reader = csv.reader(f)
    rows = [row for row in reader]

header = rows.pop(0)
data = np.float_(np.array(rows).T)

fig, ax = plt.subplots()
ax.plot(data[0], data[1], linestyle='solid', marker='o')
ax.set_xlabel(header[0])
ax.set_ylabel(header[1])

plt.show()

pandasを用いたCSVファイルの読み込みとグラフ作成

次に、「pandas」を用いてCSVデータを読み取り、matplotlibでグラフ化する方法について紹介します。pandasを用いると、文字列を数値に直す手間などが省けるため、コードを簡略化できます。

import pandas as pd
import matplotlib.pyplot as plt

path_csv = r'test\\test.csv'
df_csv = pd.read_csv(path_csv)

data_x = df_csv[df_csv.columns[0]]
data_y = df_csv[df_csv.columns[1]]

fig, ax = plt.subplots()
ax.plot(data_x, data_y, linestyle='solid', marker='o')
ax.set_xlabel(df_csv.columns[0])
ax.set_ylabel(df_csv.columns[1])

plt.show()

以上がPythonを用いてCSVファイルからデータを読み込み、グラフを作成する方法です。.