八、制图模块【ArcGIS Python系列】

import arcpy
aprx = arcpy.mp.ArcGISProject(r"C:\Projects\YosemiteNP\Yosemite.aprx") # 将引用磁盘上的 ArcGIS Pro 工程
lyt = aprx.listLayouts("Main Attractions*")[0] # 查找名称以 Main Attractions 一词开头的第一个布局
lyt.exportToPDF(r"C:\Project\YosemiteNP\Output\Yosemite.pdf", resolution=300) # 将布局导出至输出分辨率为 300 的 PDF

# 方式一
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Study.aprx") 
# 方式二 适用于程序内运行的脚本 
aprx = arcpy.mp.ArcGISProject("CURRENT")

import arcpy
aprx = arcpy.mp.ArcGISProject(r"C:\Projects\blank.aprx")
aprx.importDocument(r"C:\Projects\YosemiteNP\Documents\Yosemite.mxd")
aprx.importDocument(r"C:\Projects\YosemiteNP\Documents\Yosemite_ScenicViews.3dd")
aprx.defaultGeodatabase = r"C:\Projects\YosemiteNP\Data_Vector\YosemiteData.gdb"
aprx.defaultToolbox = r"C:\Projects\YosemiteNP\Analysis\AnalysisTools.tbx"
aprx.saveACopy(r"C:\Projects\YosemiteNP\Yosemite.aprx")

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Demo.aprx")
maps = aprx.listMaps() 
# 可以通过wild_card属性过滤，不区分大小写
# 使用maps = aprx.listMaps(“P*”)来选择P开头的地图

for m in maps:    
 print(m.name)
del aprx

import arcpy
aprx = arcpy.mp.ArcGISProject(r"C:\Projects\YosemiteNP\Yosemite.aprx")
insertLyr = arcpy.mp.LayerFile(r"C:\Projects\YosemiteNP\LayerFiles\Ranger Stations.lyrx")
m = aprx.listMaps("Yosemite National Park")[0]
refLyr = m.listLayers("Points of Interest")[0]
m.insertLayer(refLyr, insertLyr, "BEFORE")
aprx.saveACopy(r"C:\Projects\YosemiteNP\Yosemite_updated.aprx")

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Demo.aprx")
maps = aprx.listMaps()
for m in maps:    
 print("Map: " + m.name)    
 lyrs = m.listLayers()    
 for lyr in lyrs:        
  print(lyr.name)
del aprx 

import arcpy
aprx = arcpy.mp.ArcGISProject(r"C:\Projects\YosemiteNP\Yosemite.aprx")
m = aprx.listMaps("Yosemite National Park")[0]
for lyr in m.listLayers():
    if lyr.supports("DEFINITIONQUERY"):
        lyr.definitionQuery = ""
    if lyr.supports("SHOWLABELS"):
        lyr.showLabels = False
aprx.save()
del aprx

if lyr.supports("BRIGHTNESS"):
 lyr.brightness = 10

Map.addBasemap(basemap_name)

Map.addDataFromPath(data_path)

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Demo.aprx")
m = aprx.listMaps("Parks")[0]
m.addDataFromPath("C:/Mapping/Data.gdb/sidewalks")
aprx.save()
del aprx

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Demo.aprx")
m = aprx.listMaps("Parks")[0]
lyr = m.listLayers("historical_landmarks")[0]
m.removeLayer(lyr)
aprx.save()
del aprx

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Demo.aprx")
m = aprx.listMaps("Parks")[0]
lyr = m.listLayers("city_parks")[0]
m = aprx.listMaps("Facilities")[0]
map.addLayer(lyr, "BOTTOM")
aprx.save()
del aprx

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Symbology.aprx")
m = aprx.listMaps("Plan")[0]
lyrs = m.listLayers()
for lyr in lyrs:
  # 使用 Layer 对象的 symbology 属性创建一个 `Symbology` 对象
 sym = lyr.symbology
 
 # 检查图层的类型
 if lyr.isFeatureLayer:
   if hasattr(sym, "renderer"): # 图层是否支持着色器
     print(lyr.name + ": " + sym.renderer.type)
  if lyr.isRasterLayer:
   if hasattr(sym, "colorizer"): # 图层是否支持渲染器
     print(lyr.name + ": " + sym.colorizer.type)

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Symbology.aprx")
m = aprx.listMaps("Plan")[0]
lyr = m.listLayers("trails")[0]
sym = lyr.symbology
red = {"RGB": [255, 0, 0, 100]}
 if lyr.isFeatureLayer and hasattr(sym, "renderer"):
    
    # 将RGB值分配给 Symbol 对象的 color 属性
  sym.renderer.symbol.color = red
    
    # 将新符号系统应用于lyr图层的符号系统属性
  lyr.symbology = sym
    
aprx.save()
del aprx

sym.renderer.symbol.applySymbolFromGallery("Hospital", 1) # 选择第二个医院符号（从0开始）

sym.renderer.colorRamp = aprx.listColorRamps("Yellow-Orange-Red                                                  (5 Classes)")[0]

lyr.symbology = symaprx.saveACopy("Housing_Choropleth.aprx")del aprx

import arcpy
aprx = arcpy.mp.ArcGISProject("C:/Mapping/Demo.aprx")
lyt = aprx.listLayouts("Parks")[0]
mf1 = lyt.listElements("MAPFRAME_ELEMENT", "Park A")[0]
ext = mf1.camera.getExtent() # 由左下角和右上角的坐标

mf2 = lyt.listElements("MAPFRAME_ELEMENT", "Park B")[0]
mf2.camera.setExtent(ext)
aprx.save()
del aprx

import arcpy, os

# 删除已经存在的pdf
pdfPath = r"C:\Projects\YosemiteNP\AttractionsMapBook.pdf"
if os.path.exists(pdfPath):
    os.remove(pdfPath)

# 创建新 `PDFDocument` 对象
pdfDoc = arcpy.mp.PDFDocumentCreate(pdfPath)

# 将现有pdf追加到pdf对象最后一页
pdfDoc.appendPages(r"C:\Projects\YosemiteNP\Title.pdf")
pdfDoc.appendPages(r"C:\Projects\YosemiteNP\MapPages.pdf")
pdfDoc.appendPages(r"C:\Projects\YosemiteNP\ContactInfo.pdf")

# 保存 删除此对象的变量
pdfDoc.saveAndClose()
del pdfDoc

import arcpy
pdfDoc = arcpy.mp.PDFDocumentOpen(r"C:\Projects\YosemiteNP\AttractionsMapBook.pdf") 
pdfDoc.deletePages("3, 5-7") # 删除第3 5 6 7页

# 替换第3页
pdfDoc.insertPages(r"C:\Projects\Yosemite\NewPage3.pdf", 3) # 1.第二个参数前添加 2.数字是页码数 从1算起
# 替换第5-7页
pdfDoc.insertPages(r"C:\Projects\Yosemite\NewPages5-7.pdf", 5) # 插入第3页之后 页码会更新 此时在第4页后，原第8页前插入第5-7页
pdfDoc.saveAndClose()
del pdfDoc

import time, os

def export_to_png(i, lyrs, dest_folder, total_page):
    """打开指定的图层图层并导出为png
    :param i: 图层索引
    :param lyrs: 图层对象
    :param dest_folder: 导出文件夹路径
    :param total_page: 图层数量
    """
    print(f"当前处理第{i}个图层，图层名称是：", lyrs[i].name)
    
    # 设置图层可见性
    
    # 选择等于当前i索引的可见性为真 其余为假
    for j in range(total_page):
        if j == i:
            # 显示j图层
            if lyrs[j].visible == False:
                lyrs[j].visible = True
        else:
            # 关闭j图层
            if lyrs[j].visible == True:
                lyrs[j].visible = False
    
    # 导出为png
    file_name = lyrs[i].name + ".png"
    file_path = os.path.join(dest_folder, file_name)
    lyt.exportToPNG(out_png=file_path,
                   resolution=250
                   )
    print(f"{file_path}保存成功")
    

if __name__ == '__main__':
    start = time.time()
    
    # 定义工作空间
    arcpy.env.workspace = r"~\Documents\Python_\GIS\Lessons\获取哨兵2遥感地图并计算ndvi" 
       
    # 引用当前项目
    project="current" # 当前项目
    aprx = arcpy.mp.ArcGISProject(project)
    
    # 获取地图
    map_wild="波段合成" # 地图名称
    m = aprx.listMaps(map_wild)[0]
    print("当前地图是 " + m.name)
    
    # 获取图层
    lyrs = m.listLayers()
    
    # 获取图层数量 处理异常情况
    total_page = len(lyrs) # 图层数量
    if total_page!= 0:
        print("图层数量: " , total_page)
    else:
        print("错误：未找到图层") 
    
    # 获取布局
    lyt = aprx.listLayouts()[0]

    # 定义导出文件夹
    dest_folder = os.path.join(arcpy.env.workspace, "output")
    # 创建导出文件夹
    if not os.path.exists(dest_folder):
        os.makedirs(dest_folder)
    
    # 循环导出
    for i in range(total_page):
        # 调用导出函数
        export_to_png(i, lyrs, dest_folder, total_page)
        
    end = time.time()