微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

基于数据框变量创建单独的组

如何解决基于数据框变量创建单独的组

我从 Pandas 数据框中获取了 2 个具有 +100 个值的变量,现在看起来像这样:

var1 var2
0.66
0.034 没有
0.422
0.934 没有
0.66
0.04

我正在尝试对这些独立组进行 Mann whitney U 检验:

Group1 = "if var1 = "yes" | Group 2 = if var1 = "no" 

我不确定如何使这些组适合 Mann whitney U 框架?

from scipy.stats import mannwhitneyu
U1,p = mannwhitneyu(group1,group2,method="exact")

解决方法

一次性将 var2 分组并展开两个数组:

from scipy.stats import mannwhitneyu
U1,p = mannwhitneyu(*df.groupby('var2')['var1'].apply(list))
U1,p

输出:(4.0,0.4071687463160714)

其他选项:

from scipy.stats import mannwhitneyu
U1,p = mannwhitneyu(df.query('var2 == "yes"')['var1'],df.query('var2 == "no"')['var1'])
U1,p

注意。 method="exact" 参数对我不起作用,即使手动分配组

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。