您如何比较两列中的数据并选择输出哪个？

Question

我一直在用熊猫编写此代码

import pandas as pd
data = pd.read_csv('processedblock2.csv')
df = pd.DataFrame(data, columns = ['index', 'x','y','z'])#in order for this to work needed to clen up the csv file a lot


#data.columns = pd.read_csv('pdbtesttrimmed.txt')
#this sets up a boolean of true or false
add = df.x >= df.y
result =[add]
if result = True:
    data.to_csv('pdboutput2.csv')

[当我希望添加结果成为是否输入数据的条件时，它将输出到pdboutput2.csv布尔数据。

任何帮助将不胜感激！

Answer 1

 import pandas as pd 
 #creating dummy dataframe with two columns 
 df = pd.DataFrame([[1,5],[3,2],[4,6],[6,7]], columns = ['A','B'])

现在，当我们运行df.A> df.B时，它将提供布尔值

要访问真实的案例，我们只需在数据帧中传递布尔值 df [df.A> df.B]，而不是将其存储在列表中而不是检查条件。

 data = df[df.A > df.B]
 #op

 # now it can be exported..
 data.to_csv('outputfilename.csv')