Pandas 基于三种不同条件的多指标设定值

Question

objective是根据列的3个条件新建多索引列(B)

B

的条件

if B<0
  CONDITION_B='l`
elif B<-1
  CONDITION_B='L`
else
  CONDITION_B='g`

我天真地想，我们可以简单地创建两个不同的掩码并将值替换为

# Handle CONDITION_B='l` and CONDITION_B='g`

mask_2 = df.loc[:,idx[:,'B']]<0
appenddf_2=mask_2.replace({True:'g',False:'l'}).rename(columns={'A':'iv'},level=1)

然后

# CONDITION_B='L`
mask_33 = df.loc[:,idx[:,'B']]<-0.1
appenddf_2=mask_33.replace({True:'G'}).rename(columns={'A':'iv'},level=1)

正如预期的那样，这将引发错误

TypeError: sequence item 1: expected str instance, bool found

我可以知道如何处理这 3 种不同的情况吗

预期输出

ONE TWO
B   B
g   L
l   l
l   g
g   l
L   L

产生错误的代码是

import pandas as pd
import numpy as np

np.random.seed(3)
arrays = [np.hstack([['One']*2, ['Two']*2]) , ['A', 'B', 'A', 'B']]
columns = pd.MultiIndex.from_arrays(arrays)
df=  pd.DataFrame(np.random.randn(5, 4), columns=list('ABAB'))

df.columns = columns

idx = pd.IndexSlice



mask_2 = df.loc[:,idx[:,'B']]<0
appenddf_2=mask_2.replace({True:'g',False:'l'}).rename(columns={'A':'iv'},level=1)
mask_33 = df.loc[:,idx[:,'B']]<-0.1
appenddf_2=mask_33.replace({True:'G'}).rename(columns={'A':'iv'},level=1)

Answer 1

我不完全明白你想做什么，但试试这样的事情：

df = pd.DataFrame({'B': [ 0, -1, -2, -2, -1,  0,  0, -1, -1, -2]})

df['ONE'] = np.where(df['B'] < 0, 'l', 'g')
df['TWO'] = np.where(df['B'] < -1, 'L', df['ONE'])
df = df.set_index(['ONE', 'TWO'])

输出结果：

>>> df
         B
ONE TWO
g   g    0
l   l   -1
    L   -2
    L   -2
    l   -1
g   g    0
    g    0
l   l   -1
    l   -1
    L   -2

Answer 2

IIUC:

np.select() 在这种情况下是理想的：

conditions=[
    df.loc[:,idx[:,'B']].lt(0) & df.loc[:,idx[:,'B']].gt(-1),
    df.loc[:,idx[:,'B']].lt(-1),
    df.loc[:,idx[:,'B']].ge(0)
]
labels=['l','L','g']
out=pd.DataFrame(np.select(conditions,labels),columns=df.loc[:,idx[:,'B']].columns)

或

通过np.where()：

s=np.where(df.loc[:,idx[:,'B']].lt(0) & df.loc[:,idx[:,'B']].gt(-1),'l',np.where(df.loc[:,idx[:,'B']].lt(-1),'L','g'))
out=pd.DataFrame(s,columns=df.loc[:,idx[:,'B']].columns)

out的输出：

   One  Two
    B   B
0   g   L
1   l   l
2   l   g
3   g   l
4   L   L

Pandas 基于三种不同条件的多指标设定值

Pandas Multi-index set value based on three different condition

python

numpy

conditional-statements

multi-index

pandas