python绘制3d建筑

python 复制代码
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)
python 复制代码
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)
相关推荐
开发者工具分享26 分钟前
如何应对敏捷转型中的团队阻力
开发语言
行云流水剑32 分钟前
【学习记录】如何使用 Python 提取 PDF 文件中的内容
python·学习·pdf
gregmankiw32 分钟前
C#调用Rust动态链接库DLL的案例
开发语言·rust·c#
roman_日积跬步-终至千里1 小时前
【Go语言基础【20】】Go的包与工程
开发语言·后端·golang
秦少游在淮海1 小时前
C++ - string 的使用 #auto #范围for #访问及遍历操作 #容量操作 #修改操作 #其他操作 #非成员函数
开发语言·c++·stl·string·范围for·auto·string 的使用
const5441 小时前
cpp自学 day2(—>运算符)
开发语言·c++
心扬1 小时前
python生成器
开发语言·python
mouseliu1 小时前
python之二:docker部署项目
前端·python
阿蒙Amon1 小时前
06. C#入门系列【自定义类型】:从青铜到王者的进阶之路
开发语言·c#
虾球xz1 小时前
CppCon 2015 学习:CLANG/C2 for Windows
开发语言·c++·windows·学习