如何在读取 CSV 文件时将不同列分别导入为字符串列表和整数列表-Python教程-PHP中文网

如何在读取 CSV 文件时将不同列分别导入为字符串列表和整数列表

本文介绍如何在使用 csv.DictReader 读取 CSV 文件时，对指定列（如 'DNA_vol'）在导入过程中直接转换为整数列表，避免后续手动类型转换，提升代码健壮性与可读性。

本文介绍如何在使用 `csv.dictreader` 读取 csv 文件时，对指定列（如 `'dna_vol'`）在导入过程中直接转换为整数列表，避免后续手动类型转换，提升代码健壮性与可读性。

在 Python 中处理实验或生物信息学数据时，CSV 文件常包含混合类型字段：样本编号（Sample_ID）、孔位标识（Well）等应保留为字符串；而体积、浓度、循环数等数值型字段（如 DNA_vol、ddH2O_vol）则需作为整数参与后续计算。若统一以字符串形式导入，后续调用 int() 转换易引发 ValueError（例如空值、空格、非数字字符），且分散处理降低可维护性。

最佳实践是在逐行读取时即完成类型转换，而非先全部存为字符串再批量转换。以下是优化后的完整示例：

import csv

filename = open('/path/to/myfile.csv', 'r')
file = csv.DictReader(filename)

Sample_ID = []
Well = []
DNA_vol = []      # 将直接存储 int 类型
ddH2O_vol = []    # 同样为 int 列表

for col in file:
    Sample_ID.append(col['Sample_ID'])  # 字符串，原样保留
    Well.append(col['Well'])            # 字符串，原样保留
    DNA_vol.append(int(col['DNA_vol']))     # ✅ 关键：此处即时转换
    ddH2O_vol.append(int(col['ddH2O_vol'])) # ✅ 同样即时转换

filename.close()  # ⚠️ 记得关闭文件（推荐改用 with 语句，见下方说明）

✅ 优势说明：

类型转换逻辑集中、清晰，每列意图一目了然；
错误可立即定位（如某行 DNA_vol 非数字时抛出 ValueError，附带具体行号和内容）；
避免冗余内存占用（无需先存字符串再建新整数列表）。

⚠️ 重要注意事项：

异常处理不可省略：生产环境中，建议用 try/except 包裹转换逻辑，以优雅处理脏数据：

try:
    DNA_vol.append(int(col['DNA_vol'].strip()))  # .strip() 去除首尾空格
except ValueError:
    raise ValueError(f"Invalid integer in 'DNA_vol' at row {file.line_num}: '{col['DNA_vol']}'")

务必关闭文件：当前代码使用 open() 后未自动关闭。强烈推荐改用 with 语句，确保资源安全释放：

with open('/path/to/myfile.csv', 'r', newline='') as f:  # 添加 newline='' 防止 Windows 换行问题
    file = csv.DictReader(f)
    # ... 同上循环逻辑（无需手动 close）

不推荐后置 map(int, list) 方案：虽然 DNA_vol_as_int = list(map(int, DNA_vol)) 在语法上可行，但它：
- 要求所有字符串都有效，一旦存在非法值会中断整个转换；
- 无法关联原始 CSV 行号，调试困难；
- 多一次遍历，效率略低；
- 违背“尽早验证输入”的工程原则。