我有一个类似这样的日期框架:
d = {'text':['A','B'],'col1': [1, 2], 'col2': [3, 4]}
df = pd.DataFrame(data=d)
df
text col1 col2
0 A 1 3
1 B 2 3
我想按列计算平均值,并在标头“文本”上方添加标签“比率”:
'mean:' meanofcol1 meanofcol2
text col1 col2
0 A 1 3
1 B 2 3
感谢您的帮助
答案 0 :(得分:3)
仅按select_dtypes
选择数字列,将mean
和merge
计数为另一个dictionary
:
d = {**df.select_dtypes(np.number).mean().to_dict(), **{'text': 'rate: '}}
print (d)
{'col1': 1.5, 'col2': 3.5, 'text': 'rate: '}
然后使用Index.map
和MultiIndex.from_arrays
创建MultiIndex
:
df.columns = pd.MultiIndex.from_arrays([df.columns.map(d.get), df.columns])
#alternative
#df.columns = [df.columns.map(d.get), df.columns]
print (df)
rate: 1.5 3.5
text col1 col2
0 A 1 3
1 B 2 4
如果所有数字列都没有第一个,则可以采用这种方法-将文本列转换为索引,创建MultiIndex
,最后创建rename_axis
:
df = df.set_index('text')
d = df.select_dtypes(np.number).mean()
print (d)
col1 1.5
col2 3.5
dtype: float64
df.columns = pd.MultiIndex.from_arrays([df.columns.map(d.get), df.columns])
#pandas 0.24.1+
df = df.rename_axis(columns=('rate: ','text'), index=None)
#pandas bellow
#df = df.rename_axis(('rate: ','text'), axis=1).rename_axis(None)
print (df)
rate: 1.5 3.5
text col1 col2
A 1 3
B 2 4
还应动态分配text
:
df = df.rename_axis(columns=('rate: ', df.index.name), index=None)
#df = df.rename_axis(('rate: ', df.index.name), axis=1).rename_axis(None)