如何根据列的某种模式对 csv 文件的行进行分组？

Question

I have the following csv file. I want to group the rows based on the pattern of the 'city' column. If the City is X then first three rows will group into a csv file. If the City pattern is YZ then the 4th and 5th rows forms another group. Likewise, if the city pattern is YAZ then the 6, 7 and 8 rows forms a group. Again, if the pattern XX is repeated then 9th and 10th rows forms another group.

Sample of the csv file

Answer 1

If I understand correctly you are trying to filter the rows of the csv file based on the values of the last column? Here is an example using nested for loops. It might need some adjustments to match exactly what you need though.

def filter_csv(csvfile):
    combs = ["X", "XX", "YAZ", "YZ"]
    groups = [[] for _ in combs]
    lines = csvfile.split("\n")
    for line in lines:
        cells = line.split(",")
        for i, comb in enumerate(combs):
            if cells[-1] in comb:
                 groups[i].append(line)            
     return groups

 filter_csv(csvfile)