主页 / coding / 问题 / 79155135
Accepted
Leyla Elkhamlichi
Asked: 2024-11-04 18:54:27 +0800 CST

使用长数据框的条件用另一列的值填充一列

  • 772

我想用另一列的值填充数据框，我使用条件来填充值。我想要填写的条件是当列“Essentieel_Optioneel”中的值 ==“essentieel”时，需要用 perc_essentieel_skill 填充值。当值为“optioneel”时也是如此。

当我削减这个时我得到一个错误 在此处输入图片描述

conditions = [
     (df1['Essentieel_Optioneel'] <= 'essentieel'),
     (df1['Essentieel_Optioneel'] <= 'optioneel')

]

values = df1[['perc_essentieel_skill','perc_essentieel_skill']]
df1['vector'] = np.select(conditions, values)

df1

    811 'list of cases must be same length as list of conditions')
    813 # Now that the dtype is known, handle the deprecated select([], []) case
    814 if len(condlist) == 0:

ValueError: list of cases must be same length as list of conditions

我觉得我的数据框太长了，我有一个 19913 * 12 列的框。

我感觉我必须使用 for 循环。

python

1 个回答

  1. Best Answer

    为了清楚起见，您应该提供一个完整的示例，但假设您想使用这两列作为替换，则需要转置values并转换为 numpy 数组：

    values = df1[['perc_essentieel_skill1', 'perc_essentieel_skill2']].T.values
df1['vector'] = np.select(conditions, values)

    或者手动分配每个条件的列：

    df1['vector'] = np.select(conditions, [df1['perc_essentieel_skill1'],
                                       df1['perc_essentieel_skill2']])

    例子：

      Essentieel_Optioneel perc_essentieel_skill1 perc_essentieel_skill2 vector
0           essentieel                     A1                     B1     A1
1           essentieel                     A2                     B2     A2
2            optioneel                     A3                     B3     B3

    如果“Essentieel_Optioneel”中有离散类别，您还可以重构代码以使用索引查找：

    d = {'essentieel': 'perc_essentieel_skill1',
     'optioneel': 'perc_essentieel_skill2'
    }

idx, cols = pd.factorize(df1['Essentieel_Optioneel'].map(d))

df1['vector'] = df1.reindex(cols, axis=1).to_numpy()[np.arange(len(df1)), idx]
    • 0

相关问题