如何根据列表首元素将文件行归类至不同数组

错误原因分析 #

• 直接将列表mylist与整数1比较，逻辑错误：mylist是分割后的列表，需取第一个元素mylist[0]进行判断；且split()返回的是字符串类型，要和字符串"1"比较或转为整数后比较。
• 未初始化用于存储分类结果的数组，无法实现归类需求。

修正后的完整代码 #

# 初始化四个分类数组
group1 = []
group2 = []
group3 = []
group4 = []

with open('/Users/ameliajennings/Downloads/data_example.txt') as f:
    line = f.readline() 
    while line:
        # 去除行尾换行符后分割，避免多余空元素
        mylist = line.strip().split(",")
        # 确保行不为空
        if mylist:
            try:
                # 将行首元素转为整数
                first_item = int(mylist[0])
                # 根据首元素归类到对应数组
                if first_item == 1:
                    group1.append(mylist)
                elif first_item == 2:
                    group2.append(mylist)
                elif first_item == 3:
                    group3.append(mylist)
                elif first_item == 4:
                    group4.append(mylist)
            except ValueError:
                # 处理行首非数字的异常情况
                print(f"跳过无效行：{line.strip()}")
        line = f.readline()

# 验证分类结果
print("首元素为1的行集合：", group1)
print("首元素为2的行集合：", group2)
print("首元素为3的行集合：", group3)
print("首元素为4的行集合：", group4)

关键修改说明 #

1. 类型匹配：将行首元素转为整数后再比较，避免字符串与整数的类型不匹配问题；也可直接用mylist[0] == "1"进行字符串比较（需确保文件中行首无多余空格）。
2. 数据清洗：用strip()去除每行首尾的空白字符（包括换行符），防止分割后出现无效空元素。
3. 异常处理：增加try-except捕获行首无法转为整数的情况，避免程序崩溃。
4. 分类存储：初始化四个列表，分别存储对应首元素的行数据，完成归类需求。

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