def linearCombo(a, b, c):
fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(111, projection="3d")
vec = np.array(
[
[[0, 0, 0, 1, 0, 0]],
[[0, 0, 0, 0, 1, 0]],
[[0, 0, 0, 0, 0, 1]],
[[0, 0, 0, a, 0, 0]],
[[0, 0, 0, 0, b, 0]],
[[0, 0, 0, 0, 0, c]],
[[0, 0, 0, a, b, c]],
]
)
colors = ["b", "b", "b", "r", "r", "r", "g"]
for i in range(vec.shape[0]):
X, Y, Z, U, V, W = zip(*vec[i, :, :])
ax.quiver(
X,
Y,
Z,
U,
V,
W,
length=1,
normalize=False,
color=colors[i],
arrow_length_ratio=0.08,
pivot="tail",
linestyles="solid",
linewidths=3,
alpha=0.6,
)
dlines = np.array(
[
[[a, 0, 0], [a, b, 0]],
[[0, b, 0], [a, b, 0]],
[[0, 0, c], [a, b, c]],
[[0, 0, c], [a, 0, c]],
[[a, 0, c], [a, b, c]],
[[0, 0, c], [0, b, c]],
[[0, b, c], [a, b, c]],
[[a, 0, 0], [a, 0, c]],
[[0, b, 0], [0, b, c]],
[[a, b, 0], [a, b, c]],
]
)
colors = ["k", "k", "g", "k", "k", "k", "k", "k", "k"]
for i in range(dlines.shape[0]):
ax.plot(
dlines[i, :, 0],
dlines[i, :, 1],
dlines[i, :, 2],
lw=3,
ls="--",
color="black",
alpha=0.5,
)
ax.text(x=a, y=b, z=c, s=" $(%0.d, %0.d, %.0d)$" % (a, b, c), size=18)
ax.text(x=a, y=0, z=0, s=" $%0.d e_1 = (%0.d, 0, 0)$" % (a, a), size=15)
ax.text(x=0, y=b, z=0, s=" $%0.d e_2 = (0, %0.d, 0)$" % (b, b), size=15)
ax.text(x=0, y=0, z=c, s=" $%0.d e_3 = (0, 0, %0.d)$" % (c, c), size=15)
ax.grid()
ax.set_xlim([0, a + 1])
ax.set_ylim([0, b + 1])
ax.set_zlim([0, c + 1])
ax.set_xlabel("x-axis", size=18)
ax.set_ylabel("y-axis", size=18)
ax.set_zlabel("z-axis", size=18)
ax.set_title("Vector $(%0.d, %0.d, %.0d)$ Visualization" % (a, b, c), size=20)
ax.view_init(elev=20.0, azim=15)
if __name__ == "__main__":
a = 7
b = 4
c = 9
linearCombo(a, b, c)