我有一个测试分数的频率表:
score count
----- -----
77 1105
78 940
79 1222
80 4339
etc
我想显示样本的基本统计数据和箱图,由频率表汇总。 (例如,上述示例的平均值为79.16,中位数为80.)
Pandas有办法做到这一点吗?我见过的所有例子都假设了一个个案的表格。
我想我可以生成一个单独的分数列表,比如这个 -
In [2]: s = pd.Series([77] * 1105 + [78] * 940 + [79] * 1222 + [80] * 4339)
In [3]: s.describe()
Out[3]:
count 7606.000000
mean 79.156324
std 1.118439
min 77.000000
25% 78.000000
50% 80.000000
75% 80.000000
max 80.000000
dtype: float64
- 但我希望避免这种情况;真实的非玩具数据集中的总频率已达数十亿。
任何帮助表示感谢。
(我认为这是与Using describe() with weighted data不同的问题,即关于对个别情况应用权重。)
答案 0 :(得分:3)
这是一个计算频率分布的描述性统计数据的小函数:
# from __future__ import division (for Python 2)
def descriptives_from_agg(values, freqs):
values = np.array(values)
freqs = np.array(freqs)
arg_sorted = np.argsort(values)
values = values[arg_sorted]
freqs = freqs[arg_sorted]
count = freqs.sum()
fx = values * freqs
mean = fx.sum() / count
variance = ((freqs * values**2).sum() / count) - mean**2
variance = count / (count - 1) * variance # dof correction for sample variance
std = np.sqrt(variance)
minimum = np.min(values)
maximum = np.max(values)
cumcount = np.cumsum(freqs)
Q1 = values[np.searchsorted(cumcount, 0.25*count)]
Q2 = values[np.searchsorted(cumcount, 0.50*count)]
Q3 = values[np.searchsorted(cumcount, 0.75*count)]
idx = ['count', 'mean', 'std', 'min', '25%', '50%', '75%', 'max']
result = pd.Series([count, mean, std, minimum, Q1, Q2, Q3, maximum], index=idx)
return result
演示:
np.random.seed(0)
val = np.random.normal(100, 5, 1000).astype(int)
pd.Series(val).describe()
Out:
count 1000.000000
mean 99.274000
std 4.945845
min 84.000000
25% 96.000000
50% 99.000000
75% 103.000000
max 113.000000
dtype: float64
vc = pd.value_counts(val)
descriptives_from_agg(vc.index, vc.values)
Out:
count 1000.000000
mean 99.274000
std 4.945845
min 84.000000
25% 96.000000
50% 99.000000
75% 103.000000
max 113.000000
dtype: float64
请注意,这不会处理NaN并且未经过适当测试。
答案 1 :(得分:1)
在我原来的问题中,我说我不想从频率表中重建原始值,但只要它适合内存,我现在认为我会走这条路,特别是因为我的实际用例涉及更多列。
如果有人感兴趣,这是我将频率表转换为案例的功能。
In [5]: def freqs2cases(df, freq_col, cases_cols):
...: def itcases():
...: for i, row in df.iterrows():
...: for j in range(int(row[freq_col])):
...: yield row[cases_cols]
...: return pd.DataFrame(itcases())
...:
In [8]: freq_df
Out[8]:
course score freq
0 math 75 3
1 math 81 4
2 chem 92 2
3 chem 66 3
In [9]: freqs2cases(freq_df, 'freq', ['course', 'score'])
Out[9]:
course score
0 math 75
0 math 75
0 math 75
1 math 81
1 math 81
1 math 81
1 math 81
2 chem 92
2 chem 92
3 chem 66
3 chem 66
3 chem 66