[데이터 시각화] Matplotlib로 3D scatter plot 그리기

[데이터 시각화] Matplotlib로 3D scatter plot 그리기

trainset.csv

간단한 3D scatter plot을 그려보겠습니다.

정말 간단합니다. 

간단한 설명을 추가하자면 gca는 Get current axes 입니다.

비행기의 이륙 속도, 짐의 무게를 가지고 이륙에 필요한 거리를 계산하는 예제의 시각화 부분입니다.

import matplotlib.pyplot as plt
import numpy as np


fig = plt.figure()
ax = fig.gca(projection='3d')

# Make data.
data = np.loadtxt('./trainset.csv', unpack=True, delimiter=',', skiprows=1)
X = data[0]
Y = data[1]
Z = data[2]

# Plot the surface.

# Customize the z axis.
ax.scatter(X, Y, Z)

ax.set_xlabel('Speed(km/h)')
ax.set_ylabel('Weight(ton)')
ax.set_zlabel('Distance(m)')
plt.suptitle('Takeoff distance prediction', fontsize=16)
plt.show()

데이터는 우측 상단링크로 받으실 수 있습니다.

다음 포스팅에선 신경망(Multi-Layer Perceptron)을 이용하여 비행기 이륙거리 예측을 해보겠습니다.