pythonのmatplotlibで3次元の直方体を描きたい

Question

pythonのmatplotlibのグラフ上に3次元の直方体を描きたいです．調べてみたらmplot3dなどを使えば出来そうな気がしましたが，使い方がよくわかりませんでした．pythonはpython3.5を使っています．どなたか分かる方ご教授お願いします。
よろしくお願いします．

Answer 1

こんな感じでよいのでしょうか？

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt

x = [-1,1]
y = [-1,1]
X,Y = np.meshgrid(x,y)

ax = plt.subplot(111, projection='3d')
ax.plot_surface(X,Y,1,alpha=0.7)
ax.plot_surface(X,Y,-1,alpha=0.7)
ax.plot_surface(X,1,Y,alpha=0.7)
ax.plot_surface(X,-1,Y,alpha=0.7)
ax.plot_surface(-1,X,Y,alpha=0.7)
ax.plot_surface(-1,X,Y,alpha=0.7)
ax.set_xlim(-2, 2)
ax.set_ylim(-2, 2)
ax.set_zlim(-2, 2)
plt.show()

Comments

[uc200]

素早い解答ありがとうございます！
お陰様でとても助かりました。

Answer 2

numpyを使わないバージョン

from itertools import combinations
from itertools import product
from itertools import repeat

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

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

points = product(*repeat((-1, 1), 3))
for p1, p2 in combinations(points, 2):
    if [*map(sum, zip(p1, p2))].count(0) ==1:
       ax.plot3D(*zip(p1, p2), color="b")

plt.show()