我有以下代碼：x1 = df[(df['event_name'] == 'Listen') & (df['cell'] == 'CONTROL')].yes_user_county1 = df[(df['event_name'] == 'Listen') & (df['cell'] != 'CONTROL')].yes_user_countx_n = df[(df['event_name'] == 'Listen') & (df['cell'] == 'CONTROL')].total_user_county_n = df[(df['event_name'] == 'Listen') & (df['cell'] != 'CONTROL')].total_user_countzscore, pval = proportions_ztest([x1, y1], [x_n, y_n])但是，ztest 函數(shù)不起作用。原因看起來(lái)是x1、y1、x_n和y_n是熊貓系列。我嘗試使用 x1.astype(int) 但它似乎沒(méi)有解決問(wèn)題。有人可以建議什么是最好的解決方案嗎？
1 回答
哆啦的時(shí)光機(jī)
TA貢獻(xiàn)1779條經(jīng)驗(yàn) 獲得超6個(gè)贊
你想要.tolist()：
x1 = df[(df['event_name'] == 'Listen') & (df['cell'] == 'CONTROL')].yes_user_count.tolist()
y1 = df[(df['event_name'] == 'Listen') & (df['cell'] != 'CONTROL')].yes_user_count.tolist()
x_n = df[(df['event_name'] == 'Listen') & (df['cell'] == 'CONTROL')].total_user_count.tolist()
y_n = df[(df['event_name'] == 'Listen') & (df['cell'] != 'CONTROL')].total_user_count.tolist()
zscore, pval = proportions_ztest([x1, y1], [x_n, y_n])
但我也建議使用loc列名作為第二個(gè)值：
x1 = df.loc[(df['event_name'] == 'Listen') & (df['cell'] == 'CONTROL'), 'yes_user_count'].tolist()
y1 = df.loc[(df['event_name'] == 'Listen') & (df['cell'] != 'CONTROL'), 'yes_user_count'].tolist()
x_n = df.loc[(df['event_name'] == 'Listen') & (df['cell'] == 'CONTROL'), 'total_user_count'].tolist()
y_n = df.loc[(df['event_name'] == 'Listen') & (df['cell'] != 'CONTROL'), 'total_user_count'].tolist()
zscore, pval = proportions_ztest([x1, y1], [x_n, y_n])
或者甚至更好：
mask1 = (df['event_name'] == 'Listen') & (df['cell'] == 'CONTROL')
mask2 = (df['event_name'] == 'Listen') & (df['cell'] != 'CONTROL')
x1 = df.loc[mask1, 'yes_user_count'].tolist()
y1 = df.loc[mask2, 'yes_user_count'].tolist()
x_n = df.loc[mask1, 'total_user_count'].tolist()
y_n = df.loc[mask2, 'total_user_count'].tolist()
zscore, pval = proportions_ztest([x1, y1], [x_n, y_n])
添加回答
舉報(bào)
0/150
提交
取消