如何解决编写一个函数,对数据帧列表执行多个学生 t 检验
我有这个数据帧:
print(TempvsDType)
CurrentThermostatTemp
DwellingType
Bungalow 0.0
Bungalow 22.0
Bungalow 22.0
Bungalow 25.0
Bungalow 18.0
Bungalow 21.0
Bungalow 22.0
Bungalow 10.0
Bungalow 18.0
Bungalow 20.0
Bungalow 20.0
Bungalow 22.0
Bungalow 20.0
Bungalow 10.0
Bungalow 30.0
Bungalow 22.0
Bungalow 20.0
Bungalow 20.0
Bungalow 19.0
Bungalow 20.0
Bungalow 22.0
Bungalow 20.0
Bungalow 21.0
Bungalow 22.0
Bungalow 15.0
Bungalow 22.0
Bungalow 0.0
Bungalow 24.0
Bungalow 30.0
Bungalow 20.0
... ...
Park Home 20.0
Park Home 23.0
Park Home 20.0
Park Home 20.0
Park Home 20.0
Park Home 18.0
Park Home 20.0
Park Home 15.0
Park Home 12.0
Park Home 20.0
Park Home 20.0
Park Home 23.0
Park Home 21.0
Park Home 20.0
Park Home 20.0
Park Home 20.0
Park Home 23.0
Park Home 18.0
Park Home 20.0
Park Home 18.0
Park Home 16.0
Park Home 17.0
Park Home 20.0
Park Home 20.0
Park Home 18.0
Park Home 18.0
Park Home 20.0
Park Home 20.0
Park Home 15.0
Park Home 21.0
[6247 rows x 1 columns]
Flat = TempvsDType.truncate(before="Flat",after="Flat")
House = TempvsDType.truncate(before="House",after="House")
Bungalow = TempvsDType.truncate(before="Bungalow",after="Bungalow")
Maisonette = TempvsDType.truncate(before="Maisonette",after="Maisonette")
ParkHome = TempvsDType.truncate(before="Park Home",after="Park Home")
我的目标是对变量之间的所有可能组合执行学生 t 检验,重复项或重复对除外。但是,我不得不手动执行此操作,这非常耗时,尤其是对于其他变量超过 5 个且组合数量大幅增加的脚本。这是我的手动方法:
from scipy.stats import ttest_ind
#All possible combinations:
Flat_House = ttest_ind(Flat,House)
Flat_Bungalow = ttest_ind(Flat,Bungalow)
Flat_Maisonette = ttest_ind(Flat,Maisonette)
Flat_ParkHome = ttest_ind(Flat,ParkHome)
House_Bungalow = ttest_ind(House,Bungalow)
House_Maisonette = ttest_ind(House,Maisonette)
House_ParkHome = ttest_ind(House,ParkHome)
Bungalow_Maisonette = ttest_ind(Bungalow,Maisonette)
Bungalow_ParkHome = ttest_ind(Bungalow,ParkHome)
Maisonette_ParkHome = ttest_ind(Maisonette,ParkHome)
#t-test between each combination
print("t-test between {} and {} is {} and p-value:{}".format(u[0],u[1],Flat_House[0],Flat_House[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[0],u[2],Flat_Bungalow[0],Flat_Bungalow[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[0],u[3],Flat_Maisonette[0],Flat_Maisonette[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[0],u[4],Flat_ParkHome[0],Flat_ParkHome[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[1],House_Bungalow[0],House_Bungalow[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[1],House_Maisonette[0],House_Maisonette[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[1],House_ParkHome[0],House_ParkHome[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[2],Bungalow_Maisonette[0],Bungalow_Maisonette[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[2],Bungalow_ParkHome[0],Bungalow_ParkHome[1]))
print("t-test between {} and {} is {} and p-value:{}".format(u[3],Maisonette_ParkHome[0],Maisonette_ParkHome[1]))
因此,我想知道如何编写一个自动执行此操作的函数,即为除重复项和现有对之外的所有可能组合打印学生 t 检验,并以我手动打印的方式返回它。我已经尝试了很多次,但都没有成功。如果有人可以帮助我,我会很高兴。谢谢。
解决方法
from itertools import combinations
from scipy.stats import ttest_ind
dfs = dict(tuple(TempvsDType.drop_duplicates(inplace=False).groupby('DwellingType'))) # drop duplicate rows,and create a dictionary of dataframes after grouping by DwellingType
def ttest(pair):
results= ttest_ind(dfs[pair[0]]['CurrentThermostatTemp'],dfs[pair[1]]['CurrentThermostatTemp'])
print(f"t-test between {pair[0]} and {pair[1]} is {results[0]} and p-value: {results[1]}")
all_combinations = list(combinations(list(dfs.keys()),2)) # find all combinations in the keys of the dict with dataframes
[ttest(i) for i in all_combinations] # pass all combinations through the function ttest
输出:
t-test between Bungalow and Park Home is 0.2594309721800956 and p-value: 0.7984182890048678
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。