时间戳

首页 > 解决方案 > 如何在google colab上循环编写和保存csv文件？

python - 如何在google colab上循环编写和保存csv文件？

问题描述

我正在尝试在 google colab notebook 上创建和保存多个 csv 文件（循环中，每次迭代 1 个文件）。我在用着

drive.mount('/drive')
f3.to_csv('/drive/My Drive/Colab Notebooks/data/file.csv', index =False)

但显然它不起作用。单元格给了我一个文本框来输入我的授权码。有没有办法在循环中保存文件？

为了清楚起见，这是代码

for k in portkey:
 f3 = f2[f2.port_key_anonymised == k]
 if f3.shape[0]>0 :
  drive.mount('/drive')
  f3.to_csv('/drive/My Drive/Colab Notebooks/data/'+str(nb)+'.csv', index =False)

谢谢！

标签： pythoncsvsavegoogle-colaboratory

解决方案

您只需提供授权码即可安装驱动器一次。之后，您可以访问文件系统并可以读取和写入多个文件。

你的代码应该是这样的

# Mount drive
drive.mount('/drive')

# Save multiple CSV files
for k in portkey:
 f3 = f2[f2.port_key_anonymised == k]
 if f3.shape[0]>0 :
  # Save file
  f3.to_csv('/drive/My Drive/Colab Notebooks/data/'+str(nb)+'.csv', index =False)

推荐阅读