python - 如何通过列值唯一标识 csv 中的行，然后检查另一列是否有 "Passed"关键字并打印它

Question

我有一个 CSV 文件，

APIName_UniqueId,TC_Status,Comment
123456,PASSED,API "123456_Get Download Credential_1" is PASSED 
123457,PASSED,API "123457_Get Upload Credential_1" is PASSED 

我想检查 APIName_UniqueId 并检查注释是否有“已通过”关键字。

如果两个条件都符合，我想打印评论，否则打印失败的语句。

我尝试过这个脚本，但它不起作用，

import pandas as pd

df = pd.read_csv('C:\Work\Test\Python_Scripts\Final_Result.csv')

for value_i in df.APIName_UniqueId:

if df[(df['APIName_UniqueId'] == 123456) &(df['Comment'].str.contains('PASSED'))]:
             print(df['Comment'])

    else:
        print('Test Case Failed')

有人可以帮我吗？

Answer 1

您可以使用 np.where 传递 3 个参数:(1) 条件，(2) 如果为 True，则结果，(3) 如果为 False，则结果。

(np.where((df['APIName_UniqueId'] == 123456) & 
          (df['Comment'].str.contains('PASSED')), df['Comment'], 'Test Case Failed')
)
output:
array(['API "123456_Get Download Credential_1" is PASSED ',
       'Test Case Failed'], dtype=object)

请记住，建议尽可能避免使用 for 循环结构，它们不像 native pandas 和 numpy 函数那样进行优化