使用多个if条件与`str.format()`函数

3

我有一个如下的数据框。

我检查score1、2、3列并打印相应的科目。我可以比较两列并打印相应的文本。

我如何包含另一列?

import pandas as pd
import numpy as np

raw_data = {'Sub1':['A','B','C','D','E'],
            'Sub2':['F','G','H','I','J'],
            'Sub3':['K','L','M','N','O'],
    'S_score1': [1, 0, 0, 6,0], 
    'F_score1' : [0, 1,0,0,0],
    'L_score1' : [1,2,3,0,4],
    'S_score2': [0, 0, 0, 6,0], 
    'F_score2' : [0, 1,0,0,0],
    'L_score2' : [1,2,3,0,4],
    'S_score3': [0, 0, 0, 6,0], 
    'F_score3' : [0, 1,0,0,0],
    'L_score3' : [1,2,3,0,4]}

df2 = pd.DataFrame(raw_data, columns = ['Sub1','Sub2','Sub3','S_score1', 'F_score1','L_score1','S_score2', 'F_score2','L_score2','S_score3', 'F_score3','L_score3'])

def S_text(f):
    s_text = "You have scored on {}" .format(f['Sub1']) if f['S_score1'] >= 1 else "You have scored on {}" .format(f['Sub2'])
    return s_text

def F_text(f):
    f_text = "You have scored on {}" .format(f['Sub1']) if f['F_score1'] >= 1 else "You have scored on {}" .format(f['Sub2'])
    return f_text

def L_text(f):
    l_text = "You have scored on {}" .format(f['Sub1']) if f['L_score1'] >= 1 else "You have scored on {}" .format(f['Sub2'])
    return l_text

df2['s_text'] = df2.apply(S_text, axis=1)
df2['f_text'] = df2.apply(F_text, axis=1)
df2['l_text'] = df2.apply(L_text, axis=1)

我想进行如下类型的比较,但是出现了错误。基本上,如果两列满足条件(scores>=1),我想打印出这两个学科。如果三列满足条件(scores>=1),我想以以下文本形式打印出这三个学科。有没有其他方法来比较三列并打印文本?
def S_text(f):
    s_text = "You have scored on {}" .format(f['Sub1']) if f['S_score1'] >= 1 
    elif  f['S_score2'] >= 1 "You have scored on {}" .format(f['Sub2']) 
    elif f['S_score3'] >=1 "You have scored on {}" .format(f['Sub3']) 
    elif f['S_score1'] >=1 and f['S_score2']>=1 "You have scored on {} {}" .format(f['Sub1'], f['Sub2'])
    elif f['S_score1'] >=1 and f['S_score3']>=1 "You have scored on {} {}" .format(f['Sub1'], f['Sub3'])
    elif f['S_score2'] >=1 and f['S_score3']>=1 "You have scored on {} {}" .format(f['Sub2'], f['Sub3'])
    elif f['S_score1'] >=1 and f['S_score2']>=1 and f['S_score3']>=1 "You have scored on {} {} {}" .format(f['Sub1'],f['Sub2'], f['Sub3'])
    return s_text

期望输出:

输出


为什么要标记Python2和3?你是同时使用两个版本吗?你的问题是否与在一个版本上工作而在另一个版本上不工作有关,因此需要同时使用两个版本? - Patrick Artner
任何版本我都可以接受,所以我标记了两个。 - Kumar AK
我有未标记的2.7版本。谢谢。 - Kumar AK
這裡的邏輯不太清晰,能否請您解釋得更詳細一些? - BENY
预期输出是什么? - Setop
显示剩余3条评论
1个回答

0

您的数据框与预期结果不符,应该像这样才能匹配您的输出:

raw_data = {'Sub1':['A','B','C','D','E'],
            'Sub2':['F','G','H','I','J'],
            'Sub3':['K','L','M','N','O'],
    'S_score1': [1, 0, 0, 6,0], 
    'F_score1' : [0, 1,0,0,0],
    'L_score1' : [1,2,3,0,4],
    'S_score2': [0, 1, 0, 6,0], 
    'F_score2' : [0, 1,0,0,0],
    'L_score2' : [1,2,3,0,4],
    'S_score3': [0, 1, 0, 6,0], 
    'F_score3' : [0, 1,0,0,0],
    'L_score3' : [1,2,3,0,4]}

df2 = pd.DataFrame(raw_data, columns = ['Sub1','Sub2','Sub3','S_score1', 'F_score1','L_score1','S_score2', 'F_score2','L_score2','S_score3', 'F_score3','L_score3'])

我添加了一个循环来处理不同的标签。这次匹配输出。
import re
letters = ['S', 'F', 'L']
for letter in letters:
    for row in range(0, len(df2)):
        df_ = df2[[letter+'_score1', letter+'_score2', letter+'_score3']].iloc[row:row+1,0:3]
        row_string = ''
        for col in df_.columns:
            if df_[col][row] >= 1:
                row_string = row_string + ', '+ str(df2.iloc[row][df_.columns.get_loc(col)])
            row_string = re.sub(r'^,', '', row_string)
        if row_string == '':
            df2.loc[row:,letter+'_text'] = 'You have not Scored any subject'
        else:
            df2.loc[row:,letter+'_text'] = 'You have scored on' + row_string
display(df2))


    Sub1    Sub2    Sub3    S_score1    F_score1    L_score1    S_score2    F_score2    L_score2    S_score3    F_score3    L_score3    S_text  F_text  L_text
0   A   F   K   1   0   1   0   0   1   0   0   1   You have scored on A    You have not Scored any subject You have scored on A, F, K
1   B   G   L   0   1   2   1   1   2   1   1   2   You have scored on G, L You have scored on B, G, L  You have scored on B, G, L
2   C   H   M   0   0   3   0   0   3   0   0   3   You have not Scored any subject You have not Scored any subject You have scored on C, H, M
3   D   I   N   6   0   0   6   0   0   6   0   0   You have scored on D, I, N  You have not Scored any subject You have not Scored any subject
4   E   J   O   0   0   4   0   0   4   0   0   4   You have not Scored any subject You have not Scored any subject You have scored on E, J, O

我相信还有更简单的方法来做到这一点。


这在其他数据框中不起作用。我在这一行上得到了KeyError:0 - Kumar AK
这仅适用于您提供的数据框中。基本上,它是使用 scores 列创建的 de 中的列号,并将其应用于原始数据框,其中前三列是要获取信息的列。您需要将其调整为我提供的代码。此外,请将输入和输出数据框添加到问题中。我假设您正在使用提供的那个数据框。 - Jorge
@KumarAK 预期输出与S_scores数据框中的数据不匹配。请检查我回答中的编辑。 - Jorge

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接