查找值大於列平均值的列

Question

如果行值大於列的平均值（或中值），那麼如何打印列標題。 。

對於例如， df = a b c d 0 12 11 13 45 1 6 13 12 23 2 5 12 6 35

the output should be 0: a, c, d. 1: a, b, c. 2: b. 

Answer 1

In [22]: df.gt(df.mean()).T.agg(lambda x: df.columns[x].tolist()) 
Out[22]: 
0 [a, c, d] 
1  [b, c] 
2   [d] 
dtype: object 

或：

In [23]: df.gt(df.mean()).T.agg(lambda x: ', '.join(df.columns[x])) 
Out[23]: 
0 a, c, d 
1  b, c 
2   d 
dtype: object 

Answer 2

使用df.apply生成一個面罩，然後你就可以遍歷和索引df.columns：

mask = df.apply(lambda x: x > x.mean()) 
out = [(i, ', '.join(df.columns[x])) for i, x in mask.iterrows()] 
print(out) 
[(0, 'a, c, d'), (1, 'b, c'), (2, 'd')] 

Answer 3

您可以通過使用pandas試試這個，我打破了步驟

df=df.reset_index().melt('index') 
df['MEAN']=df.groupby('variable')['value'].transform('mean') 
df[df.value>df.MEAN].groupby('index').variable.apply(list) 

Out[1016]: 
index 
0 [a, c, d] 
1  [b, c] 
2   [d] 
Name: variable, dtype: object 

Answer 4

d = defaultdict(list) 
v = df.values 
[d[df.index[r]].append(df.columns[c]) 
for r, c in zip(*np.where(v > v.mean(0)))]; 
dict(d) 

{0: ['a', 'c', 'd'], 1: ['b', 'c'], 2: ['d']}