如何自动从CSV文件导入数据生成Python列表替代手动编写

用内置csv模块实现（无需额外安装库） #

如果你的CSV文件结构简单（比如单列数据、带表头），直接用Python自带的csv模块就能替代手动写列表的麻烦：

示例CSV（data.csv） #

value
AAA
BBB
CCC

替换手动列表的代码 #

import csv

my_list = []
# 打开CSV文件，utf-8编码避免乱码
with open('data.csv', 'r', encoding='utf-8') as file:
    reader = csv.reader(file)
    # 跳过表头行（如果CSV没有表头，删掉这行）
    next(reader)
    # 逐行读取，把每个值加入列表
    for row in reader:
        # 假设每行只有一个值，取第一个元素
        my_list.append(row[0])

# 接下来直接用my_list做后续操作即可
print(my_list)  # 输出: ['AAA', 'BBB', 'CCC']

如果CSV是多列数据，比如：

name,id
AAA,001
BBB,002

要提取第一列就保持row[0]，提取第二列改成row[1]，按需调整即可。

用pandas处理复杂CSV #

如果你的CSV有复杂格式（比如合并单元格、多表头），用pandas更省心，先安装依赖：

pip install pandas

然后代码更简洁：

import pandas as pd

# 读取CSV文件
df = pd.read_csv('data.csv')
# 把指定列转成列表，这里假设列名为'name'
my_list = df['name'].tolist()

# 后续直接使用列表
print(my_list)  # 输出: ['AAA', 'BBB']

这样每次只要更新CSV文件，运行代码就能自动导入最新数据，再也不用手动写['AAA', 'BBB']这种硬编码的列表了。

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