PythonのNumPyを利用して散布図を描画する方法を現役エンジニアが解説【初心者向け】

PythonのNumPyを利用して散布図を描画する方法について、TechAcademyのメンター（現役エンジニア）が実際のコードを使用して、初心者向けに解説します。

Pythonについてそもそもよく分からないという方は、Pythonとは何なのか解説した記事を読むとさらに理解が深まります。

 

なお本記事は、TechAcademyのオンラインブートキャンプ、Python講座の内容をもとに紹介しています。

 

 

目次

1. NumPyとは
2. 散布図を描画する方法
3. 実際に書いてみよう
4. まとめ

 

NumPyとは

Numpyとは、Pythonで数学的な計算を簡単に行うためのライブラリです。

例えば平均や分散、標準偏差といった統計量の計算や、sinやcosなどの三角関数の計算が簡単にできます。Numpyでは通常のPythonのリストとは異なり、ndarrayという独自のデータ型の配列が用いられるのが特徴です。

このndarrayはブロードキャストという便利な特性がある点が普通のリストと大きく異なります。この特性があるために機械学習やディープラーニングでもNumpyが好んで使われています。

Numpyは以下のようにnpという名前をつけてインポートされるのが一般的です。

import numpy as np

 

[PR] Pythonで挫折しない学習方法を動画で公開中

散布図を描画する方法

散布図を描くには、Matplotlibというライブラリのpyplotモジュールを用います。こちらを用いれば、通常の多項式や指数関数などのグラフはもちろんのこと、散布図やヒストグラムなども容易に描くことができます。また、凡例や軸ラベルを付けたり線の色を変えたりすることも可能です。

pyplotモジュールは以下のようにpltという名前をつけてインポートされるのが一般的です。

import matplotlib.pyplot as plt

 

実際に書いてみよう

それではNumpyとMatplotlibを使って実際に散布図を書いてみましょう。概要は以下の通りです。散布図の作成と表示はそれぞれ別なので注意してください。

1. 必要なライブラリをインポート
2. 標準正規分布に従った乱数を100個ずつ生成してそれぞれxとyに格納
3. x軸とy軸にラベル付け
4. 散布図を作成
5. 散布図を表示

import numpy as np

import matplotlib.pyplot as plt

#標準正規分布に従った乱数を100個生成

x = np.random.randn(100)

y = np.random.randn(100)

#x軸とy軸にラベル付け

plt.xlabel("x")

plt.ylabel("y")

#散布図を作成

plt.scatter(x, y)

#散布図を表示

plt.show()

このように綺麗な散布図を描くことができました。

まとめ

この記事ではPythonのNumPyを利用して散布図を描画する方法について解説しました。

Numpyは数学的な処理をプログラミングで行うための基本となるライブラリです。必ず使いこなせるようになりましょう。

監修してくれたメンター

柴山真沙希（しばやままさき） 大手IT企業などでエンジニアとして2年ほど勤務した後、個人事業主としてプログラミングスクール「エンペサール」を経営。子供から大人まで幅広い層を対象にプログラミングを教えている。 得意言語はPython, HTML, CSSで、機械学習やデータ分析、スクレイピングなどが得意。サッカー観戦や読書が趣味である。

 

 

TechAcademyでは、初心者でも、Pythonを使った人工知能（AI）や機械学習の基礎を習得できる、オンラインブートキャンプを開催しています。

また、現役エンジニアから学べる無料体験も実施しているので、ぜひ参加してみてください。