python - 使用 numpy loadtxt 将 CSV 文件导入 Google Colab

Question

我正在尝试将 JupyterLab 笔记本迁移到 Google Colab。在 JupyterLab 中，当我将笔记本文件和相关的 csv 文件放在同一目录中时，可以使用 numpy 的 loadtxt 函数轻松导入数据，如下所示:

import numpy as np
filein = "testfile.csv"
data = np.loadtxt(open(filein, "rb"), delimiter=",", skiprows=1)

出于各种原因，我想继续在 Colab 中使用 np.loadtxt。但是，当我在那里尝试相同的代码时，它找不到 csv 文件，尽管它位于与笔记本文件相同的 Google 云端硬盘位置。我收到此错误:"FileNotFoundError: [Errno 2] No such file or directory: 'testfile.csv'"。

我想我不知何故需要提供文件的路径，但一直无法弄清楚如何做到这一点。有没有直接使用 np.loadtxt 的方法？

Answer 1

Colab 不会自动装载 Google 云端硬盘。默认情况下，工作目录是临时后端虚拟机上的 /content。

要访问您在云端硬盘中的文件，您需要先使用以下代码片段装载它:

from google.colab import drive
drive.mount('/content/gdrive')

然后，%cd/content/gdrive/My\Drive 将工作目录更改为您的云端硬盘根目录。 (或者，根据需要自定义路径到 testfile.csv 所在的位置。)