如何使用idxmax输出子集数据帧列？

Question

我有一个数据框：

import numpy as np
import pandas as pd
df = pd.DataFrame(np.random.randint(0,40,size=(10,4)), columns=range(4), index = range(10))
df.head()

    0   1   2   3
0  27  10  13  21
1  25  12  23   8
2   2  24  24  34
3  10  11  11  10
4   0  15   0  27

我使用idxmax函数来获取包含最大值的列。

df_max = df.idxmax(1)
df_max.head()

0    0
1    0
2    3
3    1
4    3

如何使用df_max和df一起创建一个时间序列值，对应于df每行中的最大值？这是我想要的输出：

0    27
1    25
2    34
3    11
4    27
5    37
6    35
7    32
8    20
9    38

我知道我可以通过df实现这一点。max（1），但我想知道如何通过使用df_max获得相同的输出，因为我想能够将df_max应用于其他矩阵（不是df），它们与df共享相同的列和索引（但不是相同的值）。

Answer 1

你可以试试df。查找

df.lookup(df_max.index, df_max)

Out[628]: array([27, 25, 34, 11, 27], dtype=int64)

如果需要Series/DataFrame，可以将输出传递给Series/DataFrame构造函数

pd.Series(df.lookup(df_max.index, df_max), index=df_max.index)

Out[630]:
0    27
1    25
2    34
3    11
4    27
dtype: int64