그래프 생성 함수 관련 질의드립니다.

2021-08-27T11:21:16.660Z

0

아래와 같습니다. 강사님

(다음부터 질문드릴 일이 있을 때 이렇게 올리겠습니다. )

- 코드

def show_column_hist_by_target(df, column, is_amt=False):    

    cond1 = (df['TARGET'] == 1)

    cond0 = (df['TARGET'] == 0)

    fig, axs = plt.subplots(figsize=(12, 4), nrows=1, ncols=2, squeeze=False)

    # is_amt가 True이면 < 500000 조건으로 filtering

    cond_amt = True

    if is_amt:

        cond_amt = df[column] < 500000

    sns.violinplot(x='TARGET', y=column, data=df[cond_amt], ax=axs[0][0] )

    sns.distplot(df[cond0 & cond_amt][column], ax=axs[0][1], label='0', color='blue')

    sns.distplot(df[cond1 & cond_amt][column], ax=axs[0][1], label='1', color='red')

# show_column_hist_by_target(app_train, 'AMT_INCOME_TOTAL', is_amt=True)

# is_amt가 False인 경우를 테스트해보려고 했습니다.

show_column_hist_by_target(app_train, 'AMT_INCOME_TOTAL', is_amt=False)

- 오류메세지

0

음, is_amt=False일 때 cond_amt = True가 되는데, cond_amt=True가 boolean indexing으로 더 이상 먹히지 않는 군요. 

cond_amt = df[column] > 0 로 변경이 되어야 할 것 같습니다.  아래와 같이 해당 함수를 참조 부탁드립니다. 

def show_column_hist_by_target(df, column, is_amt=False):    

    cond1 = (df['TARGET'] == 1)

    cond0 = (df['TARGET'] == 0)

    fig, axs = plt.subplots(figsize=(12, 4), nrows=1, ncols=2, squeeze=False)

    # cond_amt = True가 먹히지 않음. 

    cond_amt = df[column] > 0 

    if is_amt:

        cond_amt = df[column] < 500000

    sns.violinplot(x='TARGET', y=column, data=df[cond_amt], ax=axs[0][0] )

    sns.distplot(df[cond0 & cond_amt][column], ax=axs[0][1], label='0', color='blue')

    sns.distplot(df[cond1 & cond_amt][column], ax=axs[0][1], label='1', color='red')

show_column_hist_by_target(app_train, 'AMT_INCOME_TOTAL', is_amt=False)

0

감사합니다! 강사님