python-3.x - 由 userId 指示的具有最大行数的数据集

Question

我有一个这样的数据框:

ID date        var1 var2 var3
AB 22/03/2020  0     1   3 
AB 29/03/2020  0     3   3 
CD 22/03/2020  0     1   1

我想要一个新的数据集，如果它是一个最大列(也可能发生并列)，则在行上留下与原始数据集相同数量的数据；否则设置 -1 如果它不是最大值。所以它会是:

ID date        var1   var2    var3
AB 22/03/2020  -1     -1        3 
AB 29/03/2020  -1      3        3 
CD 22/03/2020  -1      1        1

但我根本没有管理如何做到这一点，我也找不到它，我感谢任何帮助!

Answer 1

通过 DataFrame.select_dtypes 仅选择数字列:

df1 = df.select_dtypes(np.number)

或者按位置选择所有没有前两个的列 DataFrame.iloc :

df1 = df.iloc[:, 2:]

或者选择带有 var 的列标签由 DataFrame.filter :

df1 = df1.filter(like='var')

然后通过 DataFrame.where 设置新值与 max :

df[df1.columns] = df1.where(df1.eq(df1.max(1), axis=0), -1)
print (df)
   ID        date  var1  var2  var3
0  AB  22/03/2020    -1    -1     3
1  AB  29/03/2020    -1     3     3
2  CD  22/03/2020    -1     1     1