python - 对特定列中的行中的特定整数值进行计数

Question

我有 pandas 数据框，有 12 列，每列的值随机为 0 到 12。我想创建新列，其中包含这些特定值的计数。例如 newvar1 将在每行中包含“1”的计数。另一个新的结果变量将包含某些特定列中某些特定值的计数

我的数据框是这样的

v1 v2 v3 v4 
0   1  2  1 
2   3  1  1 

我想要的输出就像

v1 v2 v3 v4 newvar1_count_of_1 newvar1_count_of_1_ in_first_2_col
0   1  2  1       2                            1
2   3  1  1       2                            0

Answer 1

你只应该这样做:

df1['newvar1_count']=df1[df1.eq(1)].sum(axis=1)
df1['newvar1_count_of_1_ in_first_2_col']=df1[df1.eq(1)].loc[:,['v1','v2']].sum(axis=1)

应用到您的代码中:

df1['newvar1_count'] = df1.eq(1).sum(axis=1)
df1['newvar1_count_of_1_ in_first_2_col']=df1[df1.eq(1)].loc[:,['v1','v2']].sum(axis=1)
df1

输出:

    v1  v2  v3  v4  newvar1_count   newvar1_count_of_1_ in_first_2_col
  0 0   1   2   1   2.0             1.0
  1 2   3   1   1   2.0             0.0