如何用csv.DictReader将CSV数据同时导入字符串与整数列表?
如何将CSV读取的数值列转为整数列表?
我用csv.DictReader读取CSV文件并把数据导入多个列表,但所有列表都是字符串类型。其中Sample_ID和Well需要保持字符串列表,DNA_vol和ddH2O_vol需要转为整数列表。
当前代码:
filename = open('/path/to/myfile.csv', 'r') file = csv.DictReader(filename) Sample_ID = [] Well = [] DNA_vol =[] ddH2O_vol = [] for col in file: Sample_ID.append(col['Sample_ID']) Well.append(col['Well']) DNA_vol.append(col['DNA_vol']) ddH2O_vol.append(col['ddH2O_vol'])
生成的列表示例:
Well = ['A1', 'A2', 'A3', 'A4', 'A5'] DNA_vol = ['2', '4', '6', '8', '10', '1', '2', '4', '8', '10']
我尝试过用int()转换但没成功,请问怎么修改DNA_vol.append和ddH2O_vol.append语句让这两个列表变成整数列表?另外我不能用with open的方式。
解决方案
直接在append操作时用int()包裹对应字段即可,若CSV中存在数值前后带空格或空值的情况,可先做清理处理:
基础转换版本
filename = open('/path/to/myfile.csv', 'r') file = csv.DictReader(filename) Sample_ID = [] Well = [] DNA_vol =[] ddH2O_vol = [] 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']))
容错处理版本(应对空值/带空格的数值)
如果CSV里的数值字段存在首尾空格、空字符串或非数字内容,可添加预处理逻辑避免报错:
filename = open('/path/to/myfile.csv', 'r') file = csv.DictReader(filename) Sample_ID = [] Well = [] DNA_vol =[] ddH2O_vol = [] for col in file: Sample_ID.append(col['Sample_ID']) Well.append(col['Well']) # 先去除首尾空白,空值则默认设为0(可根据需求调整默认值) dna_val = col['DNA_vol'].strip() DNA_vol.append(int(dna_val) if dna_val else 0) h2o_val = col['ddH2O_vol'].strip() ddH2O_vol.append(int(h2o_val) if h2o_val else 0)
失败原因分析
之前转换失败大概率是因为CSV中的数值字段带有首尾空格,或者存在空字符串,通过strip()清理后再转int()即可解决大部分问题。
内容的提问来源于stack exchange,提问作者aLiveDiatribe
相关产品推荐
相关产品推荐

