Python - 比较 .csv 文件所有行中的最后一列

Question

我已从 .csv 文件中读取所有行，如下所示：-

from csv import reader, writer

# open file in read mode
with open('data.csv', 'r') as read_obj:

    # pass the file object to reader() to get the reader object
    csv_reader = reader(read_obj)
    # Iterate over each row in the csv using reader object
    for row in csv_reader:

现在，我尝试仅比较 csv 文件中每一行的最后一列。 csv文件中的数据是这样的：-

Row 1: 0,1,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,0,1,0,0,0,1
Row 2: 0,1,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,0,1,0,0,0,0
Row 3: 0,1,0,1,1,1,0,1,0,0,1,0,0,0,1,1,1,0,1,0,0,0,1
Row 4: 0,1,0,1,1,1,0,1,0,0,1,0,0,0,1,1,1,0,1,0,0,0,1
Row 5: 0,1,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,0,1,0,0,0,0

我只需要比较每一行最后一列的 1 和 0。在最后一列中，如果第一行有 1，第二行有 0，则打印第一行。

例如 1：

Row 1: 0,0,1,1
Row 2: 0,1,1,0

然后，我打印第 1 行

对于前 2：

Row 1: 0,0,1,1
Row 2: 0,1,1,1
Row 3: 0,1,1,0

然后，我打印第 2 行

对于这个问题的任何建议将不胜感激。谢谢！

Answer 1

使用熊猫

import pandas as pd

path = (r"data.csv")

data = pd.read_csv(path)

last_column = data.iloc[: , -1]

Answer 2

我希望这会有所帮助。它向您展示了如何：

1. 读取您的输入行，将'1' 和'0' 转换为1 和0，然后存储该行
2. 查找最后一行和倒数第二行
3. 比较这两行的最后一列

import csv

all_rows = []
with open('data.csv', newline='') as in_f:
    reader = csv.reader(in_f)

    # Read all rows and convert columns from string to int
    for row in reader:
        all_rows.append([int(col) for col in row])

last_row = all_rows[-1]
previous_row = all_rows[-2]

if previous_row[-1] == 1 and last_row[-1] == 0:
    with open('new.csv', 'w', newline='') as out_f:
        writer = csv.writer(out_f)
        writer.writerow(previous_row)

当我在 Ex 1 上运行它时：

0,0,1,1
0,1,1,0

我明白了：

0,0,1,1

在Ex 2上：

0,0,1,1
0,1,1,1
0,1,1,0

我明白了

0,1,1,1

还有一个大例子：

0,1,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,0,1,0,0,0,1
0,1,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,0,1,0,0,0,0
0,1,0,1,1,1,0,1,0,0,1,0,0,0,1,1,1,0,1,0,0,0,1
0,1,0,1,1,1,0,1,0,0,1,0,0,0,1,1,1,0,1,0,0,0,1
0,1,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1,0,1,0,0,0,0

我得到第 4 行：

0,1,0,1,1,1,0,1,0,0,1,0,0,0,1,1,1,0,1,0,0,0,1