实现三视图题目的自动生成

''' 生成对应的三视图和干扰选项，4个答案顺序不固定，返回题目顺序值 ''' import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D import numpy as np import random def draw_square(x_start, y_start, size): ''' 画一个正方形 ''' x = [x_start, x_start, x_start + size, x_start + size, x_start] y = [y_start, y_start + size, y_start + size, y_start, y_start] plt.plot(x, y,color='black',linewidth=5) def draw_combination(squares): if type(squares) == list: ''' 送入希望画的正方形的块数 :param squares: [2,3,4] :return: ''' x_start = 0 y_start = 0 size = 1 for count in squares: for _ in range(count): draw_square(x_start, y_start, size) y_start += size x_start += size y_start = 0 plt.axis('equal') else: ''' 画俯视图 ''' size = 1 for x in range(3): for y in range(3): if squares[x][y]: draw_square(y,x,size) plt.axis('equal') def gen_block(blocks): ''' 分别生成正视图，侧视图和俯视图需要的块数 :param block1: [[0, 1, 2], [2, 3, 4], [1, 1, 0]] :return: [2,3,4],[],[],[] ''' zheng = [max(blocks[:,0]),max(blocks[:,1]),max(blocks[:,2])] ce = [max(blocks[2]),max(blocks[1]),max(blocks[0])] fu = blocks!=0 ganrao = np.random.randint(0,2,(3,3)) return zheng,ce,fu,ganrao def draw(fig,view,i): num = '14'+str(i) ax = fig.add_subplot(eval(num)) draw_combination(view) ax.set_xlim(-0.2, 3.3) ax.set_ylim(-0.2, 3.3) plt.xticks([]) plt.yticks([]) #ax1.axis('off') ax.set_title(str(i),fontsize=32, fontweight='bold') return ax def answer(blocks,index): fig = plt.figure(figsize=(16, 4),dpi=100) # 设置图形大小 # 立体图 #blocks = np.array([[0, 0, 0], [2, 3, 2], [1, 1, 0]]) blocks = np.array(blocks) zheng,ce,fu,ganrao = gen_block(blocks) # 打乱四个答案的顺序 numbers = [1, 2, 3, 4] # 使用shuffle函数对列表进行乱序 random.shuffle(numbers) ax1 = draw(fig,zheng,numbers[0]) ax2 = draw(fig, ce, numbers[1]) ax3 = draw(fig, fu, numbers[2]) ax4 = draw(fig, ganrao, numbers[3]) # 调整子图间距 plt.subplots_adjust(wspace=0.15) plt.savefig(r'./answer/'+str(index)+'.png') return fig,numbers