import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d.art3d import Poly3DCollection
# 随机生成建筑块数据
def generate_building_blocks(num_blocks, grid_size=100, height_range=(5, 50), base_size_range=(10, 30)):
buildings = []
for _ in range(num_blocks):
# 随机选择建筑的底面起点
x = np.random.uniform(0, grid_size)
y = np.random.uniform(0, grid_size)
# 随机生成建筑的高度和底面大小
dx = np.random.uniform(*base_size_range)
dy = np.random.uniform(*base_size_range)
dz = np.random.uniform(*height_range)
buildings.append([x, y, dx, dy, dz])
return buildings
# 绘制建筑块
def plot_buildings(buildings, color='skyblue'):
fig = plt.figure(figsize=(8, 8))
ax = fig.add_subplot(111, projection='3d')
for building in buildings:
x, y, dx, dy, dz = building
# 每个建筑的顶点坐标
# vertices = [
# [(x, y, 0), (x + dx, y, 0), (x + dx, y + dy, 0), (x, y + dy, 0)], # 底面
# [(x, y, dz), (x + dx, y, dz), (x + dx, y + dy, dz), (x, y + dy, dz)], # 顶面
# ]
vertices = [
[(x, y, 0), (x + dx, y, 0), (x + dx, y + dy, 0), (x, y + dy, 0)], # 底面
[(x, y, dz), (x + dx, y, dz), (x + dx, y + dy, dz), (x, y + dy, dz)], # 顶面
[(x, y, 0), (x + dx, y, 0), (x + dx, y, dz), (x, y, dz)], # 侧面1
[(x + dx, y, 0), (x + dx, y + dy, 0), (x + dx, y + dy, dz), (x + dx, y, dz)], # 侧面2
[(x, y + dy, 0), (x + dx, y + dy, 0), (x + dx, y + dy, dz), (x, y + dy, dz)], # 侧面3
[(x, y, 0), (x, y + dy, 0), (x, y + dy, dz), (x, y, dz)] # 侧面4
]
# 构建面
for v in vertices:
ax.add_collection3d(Poly3DCollection([v], color=color, alpha=0.7))
# 绘制立方体的竖直边
for i in range(4):
ax.plot([vertices[0][i][0], vertices[1][i][0]],
[vertices[0][i][1], vertices[1][i][1]],
[vertices[0][i][2], vertices[1][i][2]], color=color)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_box_aspect([1, 1, 0.5]) # 控制显示比例
plt.show()
# 生成随机建筑数据
buildings = generate_building_blocks(num_blocks=50, grid_size=200)
# 绘制随机建筑
plot_buildings(buildings)
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d.art3d import Poly3DCollection
# 随机生成建筑块数据
def generate_building_blocks(num_blocks, grid_size=100, height_range=(5, 50), base_size_range=(10, 30)):
buildings = []
for _ in range(num_blocks):
# 随机选择建筑的底面起点
x = np.random.uniform(0, grid_size)
y = np.random.uniform(0, grid_size)
# 随机生成建筑的高度和底面大小
dx = np.random.uniform(*base_size_range)
dy = np.random.uniform(*base_size_range)
dz = np.random.uniform(*height_range)
buildings.append([x, y, dx, dy, dz])
return buildings
# 绘制建筑块
# def plot_buildings(buildings, color='skyblue'):
def plot_buildings(buildings, color='skyblue', edge_color='gray', alpha=0.8):
fig = plt.figure(figsize=(8, 8))
ax = fig.add_subplot(111, projection='3d')
ax.view_init(elev=50, azim=60) # 设置视角，elev 为俯仰角，azim 为方位角
for building in buildings:
x, y, dx, dy, dz = building
# 每个建筑的顶点坐标
# vertices = [
# [(x, y, 0), (x + dx, y, 0), (x + dx, y + dy, 0), (x, y + dy, 0)], # 底面
# [(x, y, dz), (x + dx, y, dz), (x + dx, y + dy, dz), (x, y + dy, dz)], # 顶面
# ]
vertices = [
[(x, y, 0), (x + dx, y, 0), (x + dx, y + dy, 0), (x, y + dy, 0)], # 底面
[(x, y, dz), (x + dx, y, dz), (x + dx, y + dy, dz), (x, y + dy, dz)], # 顶面
[(x, y, 0), (x + dx, y, 0), (x + dx, y, dz), (x, y, dz)], # 侧面1
[(x + dx, y, 0), (x + dx, y + dy, 0), (x + dx, y + dy, dz), (x + dx, y, dz)], # 侧面2
[(x, y + dy, 0), (x + dx, y + dy, 0), (x + dx, y + dy, dz), (x, y + dy, dz)], # 侧面3
[(x, y, 0), (x, y + dy, 0), (x, y + dy, dz), (x, y, dz)] # 侧面4
]
# 构建面
# for v in vertices:
# ax.add_collection3d(Poly3DCollection([v], color=color, alpha=0.7))
# 构建面，带有透明度和边框
for v in vertices:
poly = Poly3DCollection([v], facecolors=color, edgecolors=edge_color, linewidths=0.2, alpha=alpha)
ax.add_collection3d(poly)
# 绘制立方体的竖直边
for i in range(4):
ax.plot([vertices[0][i][0], vertices[1][i][0]],
[vertices[0][i][1], vertices[1][i][1]],
[vertices[0][i][2], vertices[1][i][2]], color=color)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_box_aspect([1, 1, 0.5]) # 控制显示比例
# 设置视角和比例
# ax.set_box_aspect([1, 1, 0.3])
# ax.set_axis_off() # 隐藏轴以增加美观
plt.show()
# 生成随机建筑数据
buildings = generate_building_blocks(num_blocks=50, grid_size=200)
# 绘制随机建筑
plot_buildings(buildings)
