Google Colab上传文件后如何动态获取文件名替代固定名称读取

解决方法 #

files.upload() 返回的 uploaded 是字典结构，字典的键对应上传文件的真实文件名，值为文件的二进制内容。

单文件上传场景（最常用）

如果你每次只上传一个文件，直接取字典的第一个键作为文件名即可，无需硬编码固定文件名：

from google.colab import files
uploaded = files.upload()

# 动态获取上传的文件名
filename = next(iter(uploaded.keys()))

with open(filename) as f:
    mylist = [line.rstrip('\n') for line in f]
    newlist = []
    for line in mylist:
        line = line.split(',')             
        line[1], line[2] = line[2], line[1] 
        line = ','.join(line)               
        print(line)
        newlist.append(line)

多文件上传场景

如果你一次性上传多个文件，可以遍历字典的所有键逐个处理：

from google.colab import files
uploaded = files.upload()

for filename in uploaded.keys():
    with open(filename) as f:
        # 此处放你原有对单个文件的处理逻辑
        mylist = [line.rstrip('\n') for line in f]
        newlist = []
        for line in mylist:
            line = line.split(',')             
            line[1], line[2] = line[2], line[1] 
            line = ','.join(line)               
            print(f"{filename} 处理后行：{line}")
            newlist.append(line)

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