pandas循环遍历一系列嵌套框来进行计算

xt0899hw 于 12个月前发布在其他

关注(0)|答案(1)|浏览(86)

这与另一个查询（pandas loop through a list of dataframes (df) to do calculation, results in previous df to be referenced and used for next df）有关，而关于循环一个数组（dfs）列表的更具体的细节，计算是两个连续dfs之间的依赖关系。
计算要求：DFS中的i_f、r_f1、r_f2和r_f3，i_f与阈值0.95比较（获取大于或等于的ID），三个r_fs比较为0.3（得到小于或等于的ID），四个比较结果一起确定第一个df中的合格ID，在下一个df中，做同样的事情，i_f与0.95的相同阈值进行比较，然而对于三个r_fs，来自先前DF的合格ID与0.3333比较，其他的与0.3比较，再次一起确定该DF中的合格ID，对于列表中的其余DF，依此类推。
下面是df列表示例，所有ID的df1中的预期输出为1，0，1，所有ID的df2中的预期输出为1，0，1，1，1。

df1 = pd.DataFrame({'ID':[1,2,3],
                    'i_f':[0.967385562,0.869575345,1],
                    'r_f1':[0.18878187,0.327355797,0.100753051],
                    'r_f2':[0.047237449,0.056038276,0.189434048],
                    'r_f3':[0.095283998,0.2554309,0.138240321]})
df2 = pd.DataFrame({'ID':[1,2,3,4,5],
                    'i_f':[0.985,1,0.993297332,1,1],
                    'r_f1':[0.300009355,0.331788473,0.146077926,0.167329833,0.245227094],
                    'r_f2':[0.152293038,0.06668,0.196683885,0.101269411,0.02493159],
                    'r_f3':[0.111617815,0.042016,0.175285158,0.085330897,0.238370325]})
df_lst = [df1, df2]`

字符串

pandas

来源：https://stackoverflow.com/questions/77749583/pandas-loop-through-a-list-of-dataframes-to-do-calculation

1条答案

按热度按时间

sqyvllje1#

逻辑与你之前的问题相同。阈值i_f是恒定的，所以这不是一个问题，不像r_f*，必须在每次迭代中计算。然而，阈值对于所有r_f*都是相同的：

for df in df_lst:
    # boolean mask for i_f
    m1 = df['i_f'].ge(0.95)

    # boolean masks for r_f*
    m2 = df.filter(like='r_f').le(df['ID'].map(thd).fillna(0.3).values, axis=0)

    # compute the final result
    res = (m1 & m2.all(axis=1)).astype(int)
    df['r_results'] = res

    # update the threshold dictionary for each group
    thd |= dict(zip(df['ID'], np.where(res, 0.33, 0.3)))
    print(df, end='\n\n')

# Output
   ID       i_f      r_f1      r_f2      r_f3  r_results
0   1  0.967386  0.188782  0.047237  0.095284          1
1   2  0.869575  0.327356  0.056038  0.255431          0
2   3  1.000000  0.100753  0.189434  0.138240          1

   ID       i_f      r_f1      r_f2      r_f3  r_results
0   1  0.985000  0.300009  0.152293  0.111618          1
1   2  1.000000  0.331788  0.066680  0.042016          0
2   3  0.993297  0.146078  0.196684  0.175285          1
3   4  1.000000  0.167330  0.101269  0.085331          1
4   5  1.000000  0.245227  0.024932  0.238370          1

字符串

赞(0）回复(0）举报 12个月前

我来回答

pandas循环遍历一系列嵌套框来进行计算

1条答案

相关问题

热门标签

最新问答