Python中如何将逗号分隔文本文件内容存储为指定结构的二维数组

当前编写的代码存在两个缺口，导致没法得到目标结构：

• 循环内直接把整行字符串作为单个元素追加到列表，没有把每行拆成独立的小元素
• 没有做元素类型转换，所有元素目前都是字符串格式，和要求的数字/字符串混合类型不匹配

具体修改逻辑 #

不需要改动前面读文件、拆分行的现有逻辑，只需要在循环块里补充两步操作即可：

1. 对每一行文本，按逗号做分割，同时清除每个分割结果前后附着的多余空格（原文件里逗号后面带了空格，不清理会让数值元素变成带空格的无效字符串）
2. 逐元素做类型匹配：每行的第一个元素是id，直接保留字符串格式；剩下的所有数值项，从字符串格式转成整数类型，最后把这一组处理好的元素组成子列表，追加到最终的二维数组里
3. 额外加个空行判断，避免文件末尾的换行符产生的空行触发类型转换错误

修改后完整代码 #

data= "data.txt"
dataArray= []
file = open(data, 'r')
contents=file.read()
file.close()
lines= contents.split("\n")
for line in lines:
    # 跳过文件中的空行
    if not line.strip():
        continue
    # 按逗号拆分每行，清除每个元素的前后空格
    split_result = [item.strip() for item in line.split(",")]
    # 匹配元素类型：id保留字符串，其余转数字
    current_row = [split_result[0]] + [int(num) for num in split_result[1:]]
    dataArray.append(current_row)

最终输出效果 #

运行后dataArray的结构完全和要求一致：

[["id1", 0, 0, 0], ["id2", 0, 0, 0], ["id3", 0, 0, 0]]

注意：如果原文件里的数值存在小数，把代码里的int替换成float即可。

内容的提问来源于stack exchange，提问作者Yiğit Er