如何在python中从csv文件中提取行

Question

在我的代码中，每个表达式都有表。每个表格都是csv格式。例如：tables.csv

+-----+-----+----------------------------+
| a   | b   | a - b > 0                  |
+-----+-----+----------------------------+
|  0  |  0  |             0              |
|  0  |  1  |             0              |
|  1  |  0  |             1              |
|  1  |  1  |             0              |
+-----+-----+----------------------------+

t = tx(expr,variables)
table = str(t)
with open('tables.csv', 'w') as f: 
mytable = f.write(table)
data = pandas.read_csv('tables.csv') 

现在，如果最后一列的值为0，我想删除这些行。如果值仅为1，我想保留行吗？

if (data.iloc[(data['expr'].to_numpy().astype(bool)).all() == 1]):
print('last col',(data['expr'].to_numpy().astype(bool).all()))

Answer 1

我不确定第三列需要什么，但可以很容易地选择带有条件的行的子集。只需使用以下内容。

df[df.a - df.b > 0]

或

df.loc[df.a - df.b > 0]