plot_vectors_in_3d.py

import numpy as np
# import nltk
import os, sys
from scipy import stats
# import helper
# import seaborn as sns
import matplotlib.pyplot as plt
# config, PROJECT_ROOT = helper.get_config()


soa = np.array([[0, 0, 1, 1, -2, 0], [0, 0, 2, 1, 1, 0],
                [0, 0, 3, 2, 1, 0], [0, 0, 4, 0.5, 0.7, 0]])

X, Y, Z, U, V, W = zip(*soa)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.quiver(X, Y, Z, U, V, W)
ax.set_xlim([-1, 0.5])
ax.set_ylim([-1, 1.5])
ax.set_zlim([-1, 8])
plt.show()