[수학] python scatter plot을 이용한 구 그리기

0. 목표

 - python scatter plot을 이용한 구 그리기

1. 실습

 1) library 호출

import numpy as np
import matplotlib.pyplot as plt

 2) 데이터 생성

x = np.linspace(-1, 1, 101)
y = np.linspace(-1, 1, 101)

 3) 격자 데이터 생성

X, Y = np.meshgrid(x, y)

 4) 격자 확인

fig = plt.figure()
fig.set_size_inches(15, 15)
plt.scatter(X, Y)

 5) Z 생성 함수 

  - z = 1 - (x^2 + y^2)

  - 뒤가 더 클 경우엔 nan으로 저장

def Z_fun(x, y):
    result = np.sqrt(1 - (x ** 2 + y ** 2))
    return result

 6) Z 생성

Z = Z_fun(X, Y)

 7) 그림 생성

  - Z 값이 양수만 생성되기에 - 부호를 붙인 Z도 추가하여 아랫부분을 추가

fig = plt.figure()
fig.set_size_inches(15, 15)
ax = fig.add_subplot(111, projection='3d')
ax.scatter(X, Y, Z)
ax.scatter(X, Y, -Z)