我在python中工作了一段时间,但是陷入了一个简单的问题。我必须为具有相同ID变量(masteruserid)的不同变量运行交叉表函数
pd.crosstab(data['MasterUserId'],visittime_cat)
pd.crosstab(data['MasterUserId'],week_cat)
现在我想做大约7-8次。我不想递归地调用交叉表函数,而是希望将其放入循环中并为每次迭代生成一个交叉表数据集。我尝试了这个,但是没有成功
def cross_tab(id_col,field):
col_names=['visittime_cat','week_cat','var3','var4']
for i in col_names:
'crosstab_{ }'.format(i)=pd.crosstab(id_col,i)
我想生成数据集,例如crosstab_visittime_cat,crosstab_week_cat或作为crosstab_1,crosstab_2等。
最佳答案
我是否建议将数据集存储在字典中?
def cross_tab(data_frame, id_col):
col_names=['visittime_cat','week_cat','var3','var4']
datasets = {}
for i in col_names:
datasets['crosstab_{}'.format(i)] = pd.crosstab(data_frame[id_col], data_frame[i])
return datasets
用虚拟数据集进行测试
import numpy as np
import pandas as pd
data = pd.DataFrame({'MasterUserId': ['one', 'one', 'two', 'three'] * 6,
'visittime_cat': ['A', 'B', 'C'] * 8,
'week_cat': ['foo', 'foo', 'foo', 'bar', 'bar', 'bar'] * 4,
'var3': np.random.randn(24),
'var4': np.random.randn(24)})
storage = cross_tab(data, "MasterUserId")
storage.keys()
['crosstab_week_cat', 'crosstab_var4', 'crosstab_visittime_cat', 'crosstab_var3']
storage['crosstab_week_cat']
week_cat bar foo
MasterUserId
one 6 6
three 4 2
two 2 4
[3 rows x 2 columns]
关于python - 循环中的python交叉表功能,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/29921007/