Pandas .idxmin(axis = 1)返回错误的列名值

时间:2019-07-27 11:34:48

标签: python python-3.x pandas

熊猫的价值:

Index   Bra Obr Zal Uto Str Nah Tec Ryc Hla BestP_A BestP_Amin  BestP_minA
0       461 38  44  46  49  137 324 322 162 Bra     Obr         137.0
1       442 32  30  35  39  180 322 325 180 Bra     Obr         180.0
2       152 28  23  23  30  175 335 355 206 Bra     Obr         175.0
3       38  33  68  33  49  119 223 46  46  Zal     Obr         46.0
4       36  33  203 36  46  217 253 166 170 Zal     Obr         166.0
5       35  84  38  41  54  49  175 57  141 Obr     Nah         49.0
6       34  45  71  45  59  72  207 57  60  Zal     Obr         57.0

Value from Pandas

我需要将列名称返回到列“ BestP_Amin”,Python返回错误的列名称。值索引示例0:

BestP_A = Bra, Bra = 461, Nah = 137, Tec = 324, Ryc = 322, Hla = 162

代码:

if data.at[i,'BestP_A'] == 'Bra':
    data['BestP_Amin'] = data[['Bra','Nah','Tec','Ryc','Hla']].idxmin(axis=1) 

返回值Obr,问题出在哪里? 阻止代码:

# nalezení nejleší pozice, přepočet podle výše atributů
for i in range(0,len(data.index)):
# nalezení pozice
    data['BestP_A'] = data[['Bra','Obr','Zal','Uto']].idxmax(axis=1)
#nalezení nejmenšího atributu
# VRACI CHYBNE HODNOTY
    if data.at[i,'BestP_A'] == 'Bra':
         data['BestP_Amin'] = data[['Bra','Nah','Tec','Ryc','Hla']].idxmin(axis=1)
    elif data.at[i,'BestP_A'] == 'Obr':
         data['BestP_Amin'] = data[['Obr','Nah','Tec','Ryc','Hla']].idxmin(axis=1)
    elif data.at[i,'BestP_A'] == 'Zal':
         data['BestP_Amin'] = data[['Zal','Nah','Tec','Ryc','Hla']].idxmin(axis=1)
    elif data.at[i,'BestP_A'] == 'Uto':
         data['BestP_Amin'] = data[['Uto','Nah','Tec','Ryc','Hla']].idxmin(axis=1)

    data.at[i,'BestP_minA'] = min(data.at[i,'Nah'],data.at[i,'Tec'],data.at[i,'Ryc'],data.at[i,'Hla'])
    data.at[i,'BestA_U'] = int(round(data.at[i,'BestA']*(data.at[i,'Ene']/100)*(1+(.25*data.at[i,'Sou']/100)+(.2*data.at[i,'Zku']/100)),0))

Block code in python:

3 个答案:

答案 0 :(得分:0)

我不知道您为什么使用for循环,但是如果您的目标是获取介于两者之间的列的名称(表示最小值),那么仅此行就足够了:

df['BestP_Amin'] = df[['Bra','Nah','Tec','Ryc','Hla']].idxmin(axis=1)

无需将它们与这些for语句放在if循环中。 idxmin已对每一行执行矢量化搜索,从而产生了该列。
之后,该列如下所示:

0    Nah
1    Nah
2    Bra
3    Bra
4    Bra
5    Bra
6    Bra

答案 1 :(得分:0)

嗯,循环“ For”必须用于“ If”。在“ BestP_A”中需要测试值,并更改填充“ BestP_Amin”的选择

答案 2 :(得分:0)

已解决,请使用元组

let array = [
      {
        item: "Pen",
        price: 3
      },
      {
        item: "Book",
        price: 10
      },
      {
        item: "Pen",
        price: 6
      },
      {
        item: "Book",
        price: 8
      }
    ];