如何将GeodataFrame下载到我的本地驱动器上，从Google Colab下载？

Question

我对 SHAPEFiles 和 GeoDataFrames 相当陌生，所以如果这是一个非常基本的问题，我深表歉意。

我在 Mac 上的 Google Chrome 中使用 Google Colab。 Python 版本 3.xx

我有一个现有的 Esri SHAPEFile，其中包含多边形和多多边形的 CRS 点（纬度-经度）。

我的目标是找到这些多边形和多多边形的质心，并将质心添加为现有 SHAPE 文件中的新列，然后在本地保存文件。

如何将文件保存到本地？我使用什么格式？ 谢谢。

Answer 1

import geopandas as gpd

df = gpd.read_file("polygons.shp")

df["geometry"] = df["geometry"].centroid  # assuming a projection CRS is used

df.to_file("centroids.shp", driver="Shapefile")

您需要刷新 Google Colabs 中的工作目录才能查看 shapefile。确保您也下载了所有关联的文件。

注意：您不能在一个 shapefile 中存储多个几何图形。

该脚本使用几何形状的质心创建一个新的 shapefile，但也保留所有其他列。这样您就能够关联两个数据集。

---

以下是一个可能的工作流程：

1. 创建一个新的 Jupyter 笔记本或打开现有的笔记本
2. 创建一个包含形状文件的 zip 文件包括所有关联文件
3. 上传将其复制到您的 Colabs 工作目录
   • 我在示例中将文件命名为 polygons.zip
4. 在单元格中运行以下代码：

!pip install geopandas

import os
import tempfile
import zipfile
import glob

import geopandas as gpd

OUTPUT_SHAPE_FILE = "centroids.shp"
SHAPE_FILE_ZIP = "polygons.zip"

with tempfile.TemporaryDirectory() as wd:

    with zipfile.ZipFile(SHAPE_FILE_ZIP, "r") as shape_file_zip:
        shape_file_zip.extractall(wd)

    shape_file = os.path.join(wd, SHAPE_FILE_ZIP.replace(".zip", ".shp"))

    df = gpd.read_file(shape_file)

    df["geometry"] = df["geometry"].centroid  # assuming a projection CRS is used

    df.to_file(os.path.join(wd, OUTPUT_SHAPE_FILE), driver="Shapefile")

    shape_file_name = OUTPUT_SHAPE_FILE.replace(".shp", "")

    with zipfile.ZipFile(f"{shape_file_name}.zip", "w") as output_file:

        for file in glob.iglob(f"{os.path.join(wd, shape_file_name)}.*"):
            output_file.write(file, os.path.basename(file))

5. 刷新 Google Colabs 上的工作目录，您将看到 centroids.zip
6. 下载文件 (将鼠标悬停在文件上时单击三个点）

---

PS。对于这种情况，我建议使用本地 Jupyter 笔记本和 Python 环境。