Google Colab缩进时!命令失效的条件Shell写法咨询

问题原因 #

Colab中!开头的命令是IPython提供的魔法语法，不属于Python原生语法范畴。IPython会在代码运行前单独预处理这类命令，不会遵循Python的缩进规则、条件分支逻辑，放在缩进代码块中要么无视外层判断直接执行，要么直接抛出识别错误。

可行实现方案 #

方案1：通过Python subprocess模块调用Shell命令（推荐） #

完全走Python原生执行逻辑，完美兼容条件判断、循环等所有Python语法，后续拓展解压、文件校验等操作也更方便。

import os
import subprocess

# 替换为你挂载Drive后的实际目标目录路径
TARGET_SAVE_DIR = "/content/drive/MyDrive/your_dataset_dir"
# 替换为你要下载的资源实际链接
DATA_DOWNLOAD_URL = "your_data_download_url"

if not os.path.isdir(TARGET_SAVE_DIR):
    # 执行wget下载，-P参数指定文件保存目录，check=True会在下载失败时抛出异常
    subprocess.run(
        ["wget", DATA_DOWNLOAD_URL, "-P", TARGET_SAVE_DIR],
        check=True
    )

方案2：将判断逻辑写入Shell命令直接执行 #

不需要额外导入Python模块，把目录存在性判断和下载逻辑都写成Shell语法，整段交给!执行即可，适合简单的单步Shell操作场景。

# Shell逻辑：目标目录不存在时才执行wget下载，替换对应路径和下载链接即可
! [ ! -d "/content/drive/MyDrive/your_dataset_dir" ] && wget your_data_download_url -P /content/drive/MyDrive/your_dataset_dir

注意事项 #

• 操作Google Drive前必须先挂载，挂载代码为from google.colab import drive; drive.mount('/content/drive')，否则路径无法正常识别
• 路径建议写绝对路径，避免Colab默认工作目录切换导致的路径匹配错误

内容的提问来源于stack exchange，提问作者Ofek Finkelstein