在Python/Pandas中自动获取上传CSV文件的文件名方法

解决方案 #

files.upload()返回的是一个字典，其中键就是上传的文件名，直接提取字典的键就能自动获取文件名，不用每次手动修改代码。

完整代码（单文件上传场景） #

from google.colab import files
import numpy as np
import pandas as pd
import io

# 触发文件上传组件
uploaded = files.upload()

# 自动获取上传的第一个文件名
filename = next(iter(uploaded.keys()))

# 自动读取CSV数据
data = pd.read_csv(io.BytesIO(uploaded[filename]))

多文件上传场景扩展 #

如果需要同时上传并处理多个CSV文件，可以遍历所有文件名逐个读取：

from google.colab import files
import numpy as np
import pandas as pd
import io

uploaded = files.upload()

# 遍历所有上传的文件
for filename in uploaded.keys():
    data = pd.read_csv(io.BytesIO(uploaded[filename]))
    # 在此添加对单个文件数据的处理逻辑
    print(f"成功读取文件：{filename}")

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