python - 如何使用 Google Colab python 读取/循环访问文件夹中的多个 .csv 文件，然后将每个文件分配为函数参数

Question

我目前正在使用 Google Colab 并且已经安装了我的 Google 云端硬盘。我在包含多个 .csv 文件的驱动器中有一个文件夹

例如文件夹名称:dataset

文件夹内容:data1.csv、data2.csv、data3.csv等

我想遍历文件夹中的每个文件，然后将文件作为函数参数

这是我的代码，但还是不行

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

def myfunction(data):
###function detail here###

dir = '/content/drive/dataset'

for files in dir:
  myfunction(pd.read_csv('filename'))

谢谢

Answer 1

您必须使用类似 os.listdir 的函数遍历文件。下面是一个使用此函数并防御性地检查读取的内容是否为 csv 文件的示例。我使用了 Google Colab 的 sample_data 文件夹，这样代码就可以重现了；您需要更改 dir 变量以指向您的 Google 云端硬盘文件夹。

import pandas as pd
import os

def myfunction(data):
  print(data)

dir = 'sample_data'

for file in os.listdir(dir):
  if file.endswith(".csv"):
    myfunction(file)