Flask中上传CSV后动态填充下拉框的问题求助

你的代码核心问题有两点：

1. 未正确传递CSV表头到模板：POST请求读取CSV后，没有把获取到的表头赋值给col_names，反而用了request.args.get('col_names')（初始为空），导致模板循环时遇到None触发"Nonetype object not iterable"错误。
2. 初始状态未处理空值：GET请求时col_names默认是None，模板里的循环直接遍历空值会报错。

修正后的app.py代码 #

from flask import Flask, render_template, request
import pandas as pd

app = Flask(__name__)

@app.route('/upload', methods=['GET', 'POST'])
def upload():
    # 初始化空列表，避免None引发循环错误
    col_names = []
    col1 = col2 = col3 = ""
    
    if request.method == 'POST':
        file = request.files.get('file')
        if file and file.filename.endswith('.csv'):
            # 读取CSV并提取表头转为列表
            df = pd.read_csv(file)
            col_names = df.columns.tolist()
            # 获取用户选择的下拉框值，用于回显
            col1 = request.form.get('col1', '')
            col2 = request.form.get('col2', '')
            col3 = request.form.get('col3', '')
        else:
            return 'Invalid file format'
    
    return render_template('upload.html', col_names=col_names,
                           col1=col1, col2=col2, col3=col3)

修正后的upload.html代码 #

<form method="POST" enctype="multipart/form-data">
    <label for="file">File:</label>
    <input type="file" id="file" name="file"><br />
    
    <label for="col1">Column 1:</label>
    <select id="col1" name="col1">
        {% for col in col_names %}
        <option value="{{ col }}" {% if col == col1 %}selected{% endif %}>{{ col }}</option>
        {% endfor %}
    </select>
    <br />
    
    <label for="col2">Column 2:</label>
    <select id="col2" name="col2">
        {% for col in col_names %}
        <option value="{{ col }}" {% if col == col2 %}selected{% endif %}>{{ col }}</option>
        {% endfor %}
    </select>
    <br />
    
    <label for="col3">Column 3:</label>
    <select id="col3" name="col3">
        {% for col in col_names %}
        <option value="{{ col }}" {% if col == col3 %}selected{% endif %}>{{ col }}</option>
        {% endfor %}
    </select>
    <br />
    
    <button>Upload</button>
</form>

关键修改说明 #

• 初始化col_names为空列表：避免GET请求或未上传文件时，模板循环空值报错。
• 从CSV提取表头：POST上传有效文件后，将df.columns.tolist()赋值给col_names，确保模板能拿到可迭代的表头列表。
• 移除URL参数传递表头：不需要通过URL参数传递表头，直接从后端传递到模板更可靠。
• 下拉框回显选中状态：通过{% if col == col1 %}selected{% endif %}保留用户之前的选择，提升交互体验。

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