Question

我有一些这样通过的测试数据：

import numpy as np
import pandas as pd

data = [[np.random.choice(['Tmax', 'Tmin']), np.random.choice(['USA', 'FRA', 'AUS']),
         np.random.randint(1,4), np.random.choice(['Class A', 'Class B']),np.random.randint(4)] for i in range(1000)]
df = pd.DataFrame(data, columns=['temp', 'region', 'method', 'slice', 'pass'])

我已使用np.mean函数对数据框进行了如下操作：

pv = pd.pivot_table(df, 
               index=['temp', 'method'], 
               values=['pass'],
               columns=['slice', 'region'],
              aggfunc=(np.mean))

'pass'变量的值是通过测试的平均值，以小数形式出现，例如0.45、1.5 3.2 ...，但是我想用以下函数映射这些结果

def results_mapper(v):
    num_tests = np.round(v)
    if num_tests < 2:
        return 1
    elif num_tests == 2:
        return 2
    else:
        return 3

我尝试使用pv.map(results_mapper)或map(results_mapper, pv)映射结果，但是没有运气。

样品编号： pv的输出数字可能类似于0.5, 1.1, 1.49, 1.6, 1.9, 2.1, 2.9, 3.5

将被映射到1, 1, 1, 2, 2, 2, 3, 3

如何通过保持数据透视表格式不变将映射功能应用于结果？

Answer 1

使用np.vectorize

func = np.vectorize(results_mapper)
pv.values[:,:] = func(pv.values)

输出

              pass                            
slice       Class A           Class B          
region          AUS  FRA  USA     AUS  FRA  USA
temp method                                    
Tmax 1          1.0  1.0  2.0     2.0  2.0  1.0
     2          1.0  2.0  1.0     1.0  1.0  1.0
     3          1.0  2.0  2.0     2.0  2.0  2.0
Tmin 1          2.0  1.0  2.0     2.0  1.0  2.0
     2          2.0  2.0  1.0     2.0  1.0  2.0
     3          2.0  1.0  1.0     2.0  2.0  1.0

ivot_table聚合后如何映射熊猫结果？

1 个答案:

输出